rolo-models / models_info.json
wuhp's picture
Update models_info.json
753e2c0 verified
raw
history blame
2.74 kB
[
{
"model_name": "human",
"display_name": "Human Detection Model",
"download_url": "https://raw.githubusercontent.com/wuhplaptop/human/main/Human-50e-11n.pt",
"architecture": "YOLOv11",
"training_epochs": 50,
"batch_size": 32,
"optimizer": "auto",
"learning_rate": 0.0005,
"data_augmentation_level": "Moderate",
"mAP_score": 0.91583,
"num_images": 10865,
"class_ids": {
"0": "Person"
},
"datasets_used": [
"detect-human-lg2ng_v1",
"human-detection-grmvx_v1",
"human-detection-p8c2v_v1",
"human-pysi7_v3",
"humans-ziarm_v2",
"people-4evn7-fqlf8-d887c_v3",
"people-4evn7_v2",
"person-dataset-kzsop-vemv4-h1uoh-q5vtx_v2",
"tello-olz2y_v5"
],
"class_image_counts": {
"Person": 10865
}
},
{
"model_name": "face",
"display_name": "Face Detection Model",
"download_url": "https://raw.githubusercontent.com/wuhplaptop/face-11-n/main/face2.pt",
"architecture": "YOLOv11",
"training_epochs": 75,
"batch_size": 32,
"optimizer": "auto",
"learning_rate": 0.0005,
"data_augmentation_level": "Moderate",
"mAP_score": 0.99373,
"num_images": 15349,
"class_ids": {
"0": "Face"
},
"datasets_used": [
"face-detection-mik1i_v24",
"faces-bfigz_v3",
"head-qug6h_v2"
],
"class_image_counts": {
"Face": 15349
}
},
{
"model_name": "guns",
"display_name": "Guns and Knives Detection Model",
"download_url": "https://raw.githubusercontent.com/wuhplaptop/guns-100-11m/main/Guns-100-11m.pt",
"architecture": "YOLOv11",
"training_epochs": 100,
"batch_size": 32,
"optimizer": "auto",
"learning_rate": 0.0005,
"data_augmentation_level": "Moderate",
"mAP_score": 0.85476,
"num_images": 44219, // Sum of Gun and Knife image counts
"class_ids": {
"0": "Gun",
"1": "Knife"
},
"datasets_used": [
"gun-detection-xwosb_v8",
"gun-sqvhj_v1",
"people_with_guns_2-pvcnk_v9",
"police1_v1",
"soldier-wclp7_v1",
"test-e46au_v1",
"weapon-rl8c4_v3",
"weapondetection-fdzlo_v4",
"weapons-detection-x9fnq_v3",
"wepon_detection_v1"
],
"class_image_counts": {
"Gun": 44219,
"Knife": 4038
}
}
]