titanthree's picture
Upload config.json with huggingface_hub
41e5a32 verified
Raw
History Blame Contribute Delete
1.01 kB
{
"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
}