| { | |
| "architectures": [ | |
| "YolosForObjectDetection" | |
| ], | |
| "model_type": "yolos", | |
| "input_size": [640, 640], | |
| "anchors": [ | |
| [10, 13], | |
| [16, 30], | |
| [33, 23] | |
| ], | |
| "classes": ["0", "1"], | |
| "output_layers": [ | |
| {"layer_size": [52, 52], "anchors": 3}, | |
| {"layer_size": [26, 26], "anchors": 3}, | |
| {"layer_size": [13, 13], "anchors": 3} | |
| ], | |
| "nms_threshold": 0.5 | |
| } | |