File size: 1,008 Bytes
41e5a32 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | {
"model_name": "YOLOv8 Construction Safety Detector",
"task": "object-detection",
"framework": "PyTorch",
"input_size": 640,
"classes": {
"0": "construction-safety",
"1": "helmet",
"2": "no-helmet",
"3": "no-vest",
"4": "person",
"5": "vest"
},
"num_classes": 6,
"metrics": {
"mAP50": 0.823,
"mAP50-95": 0.427,
"precision": 0.768,
"recall": 0.806
},
"class_metrics": {
"helmet": {
"precision": 0.835,
"recall": 0.906,
"mAP50": 0.919
},
"no-helmet": {
"precision": 0.714,
"recall": 0.625,
"mAP50": 0.725
},
"no-vest": {
"precision": 0.639,
"recall": 0.755,
"mAP50": 0.678
},
"person": {
"precision": 0.81,
"recall": 0.921,
"mAP50": 0.898
},
"vest": {
"precision": 0.841,
"recall": 0.822,
"mAP50": 0.892
}
},
"dataset": "Francesco/construction-safety-gsnvb",
"epochs": 50,
"batch_size": 8,
"image_size": 640
} |