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