| { | |
| "format": "page_type_classifier_v1", | |
| "architecture": "mobilenet_v3_small", | |
| "classes": [ | |
| "cover", | |
| "story_page", | |
| "annexe", | |
| "summary" | |
| ], | |
| "input": { | |
| "name": "input", | |
| "shape": [ | |
| 1, | |
| 3, | |
| 224, | |
| 224 | |
| ], | |
| "color_space": "RGB" | |
| }, | |
| "output": { | |
| "name": "logits", | |
| "shape": [ | |
| 1, | |
| 4 | |
| ] | |
| }, | |
| "preprocessing": { | |
| "resize_shortest_edge": 256, | |
| "center_crop": 224, | |
| "mean": [ | |
| 0.485, | |
| 0.456, | |
| 0.406 | |
| ], | |
| "std": [ | |
| 0.229, | |
| 0.224, | |
| 0.225 | |
| ] | |
| }, | |
| "onnxruntime_cpu_smoke": { | |
| "output_shape": [ | |
| 1, | |
| 4 | |
| ] | |
| } | |
| } | |