digit / config.json
ipd805's picture
Update config.json
da95646 verified
raw
history blame
664 Bytes
{
"name": "YOLOv8 Digit and Icon Detector",
"framework": "onnx",
"labels": [
{
"name": "digit",
"color": "#ff0000",
"attributes": []
},
{
"name": "icon",
"color": "#00ff00",
"attributes": []
}
],
"interpretation_script": "yolo",
"weights": "best1-2.onnx",
"class_to_idx": {"digit": 0, "icon": 1},
"model_parameters": {
"confidence_threshold": 0.5,
"nms_threshold": 0.4
},
"input_size": [640, 640],
"preprocessing": {
"resize": {
"width": 640,
"height": 640
},
"normalize": {
"mean": [0.485, 0.456, 0.406],
"std": [0.229, 0.224, 0.225]
}
}
}