{ | |
"framework": "pytorch", | |
"type": "detector", | |
"classes": [ | |
"bicycle_rider" | |
], | |
"pytorch": { | |
"model_file": "best.pt", | |
"input_size": [640, 640], | |
"confidence_threshold": 0.5, | |
"nms_threshold": 0.45 | |
} | |
} |
{ | |
"framework": "pytorch", | |
"type": "detector", | |
"classes": [ | |
"bicycle_rider" | |
], | |
"pytorch": { | |
"model_file": "best.pt", | |
"input_size": [640, 640], | |
"confidence_threshold": 0.5, | |
"nms_threshold": 0.45 | |
} | |
} |