rolo-models / models_info.json
wuhp's picture
Update models_info.json
6f41ebf verified
[
{
"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,
"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
}
},
{
"model_name": "car",
"display_name": "Car Detection Model",
"download_url": "https://raw.githubusercontent.com/wuhplaptop/yolocar/main/car-75e-11n.pt",
"architecture": "YOLOv11",
"training_epochs": 75,
"batch_size": 32,
"optimizer": "auto",
"learning_rate": 0.0005,
"data_augmentation_level": "Moderate",
"mAP_score": 0.88072,
"num_images": 15163,
"class_ids": {
"0": "Vehicle"
},
"datasets_used": [
"aerial-cars-rqcqh_v2",
"bikedetection-7bpwy_v2",
"car-detection-pyxz2_v4",
"cars-bytt8_v35",
"transport-rhkah_v8",
"vehiclecount_v4",
"vehicles-q0x2v-8kns4_v1"
],
"class_image_counts": {
"Vehicle": 15163
}
},
{
"model_name": "cat",
"display_name": "Cat Detection Model",
"download_url": "https://raw.githubusercontent.com/wuhplaptop/cat-yolo/main/Cat-50e-11n.pt",
"architecture": "YOLOv11",
"training_epochs": 50,
"batch_size": 32,
"optimizer": "auto",
"learning_rate": 0.0005,
"data_augmentation_level": "Moderate",
"mAP_score": 0.98567,
"num_images": 9147,
"class_ids": {
"0": "Cat"
},
"datasets_used": [
"cat-2er75_v4",
"cats-j6k8r-iakbs_v1",
"cats-j6k8r_v1",
"mickey-finder_v1"
],
"class_image_counts": {
"Cat": 9147
}
},
{
"model_name": "dog",
"display_name": "Dog Detection Model",
"download_url": "https://raw.githubusercontent.com/wuhplaptop/dog-yolo/main/dog-75e-11n.pt",
"architecture": "YOLOv11",
"training_epochs": 75,
"batch_size": 32,
"optimizer": "auto",
"learning_rate": 0.0005,
"data_augmentation_level": "Moderate",
"mAP_score": 0.99457,
"num_images": 55924,
"class_ids": {
"0": "Full",
"1": "Head"
},
"datasets_used": [
"doggo-detection-assignment-2_v2"
],
"class_image_counts": {
"Full": 27965,
"Head": 27959
}
}
]