File size: 661 Bytes
7ef9bec
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
{
  "model_name": "RitualCNN",
  "architecture": "PyTorch convolutional neural network exported to ONNX",
  "input": {
    "name": "input",
    "shape": ["batch", 1, 128, 128],
    "type": "float32",
    "preprocessing": "Convert to grayscale, resize to 128x128, threshold at 127, normalize to 0 or 1."
  },
  "output": {
    "name": "output",
    "shape": ["batch", 7],
    "type": "float32",
    "meaning": "Raw class scores (logits). Select the label at the index with the highest score."
  },
  "labels": ["Acervus", "Avarus", "Flosculus", "Mutus", "Not", "Pudor", "Regium"],
  "onnx_opset": 11,
  "source_code": "https://github.com/methil-group/rituals"
}