| { | |
| "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" | |
| } | |