Spaces:
Sleeping
Sleeping
Update models_info.json
Browse files- models_info.json +83 -7
models_info.json
CHANGED
@@ -1,13 +1,89 @@
|
|
1 |
[
|
2 |
{
|
3 |
-
"model_name": "
|
4 |
-
"
|
5 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
},
|
7 |
{
|
8 |
-
"model_name": "
|
9 |
-
"
|
10 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
}
|
12 |
-
// Add more models as needed
|
13 |
]
|
|
|
1 |
[
|
2 |
{
|
3 |
+
"model_name": "human",
|
4 |
+
"display_name": "Human Detection Model",
|
5 |
+
"download_url": "https://raw.githubusercontent.com/wuhplaptop/human/main/Human-50e-11n.pt",
|
6 |
+
"architecture": "YOLOv11",
|
7 |
+
"training_epochs": 50,
|
8 |
+
"batch_size": 32,
|
9 |
+
"optimizer": "auto",
|
10 |
+
"learning_rate": 0.0005,
|
11 |
+
"data_augmentation_level": "Moderate",
|
12 |
+
"mAP_score": 0.91583,
|
13 |
+
"num_images": 10865,
|
14 |
+
"class_ids": {
|
15 |
+
"0": "Person"
|
16 |
+
},
|
17 |
+
"datasets_used": [
|
18 |
+
"detect-human-lg2ng_v1",
|
19 |
+
"human-detection-grmvx_v1",
|
20 |
+
"human-detection-p8c2v_v1",
|
21 |
+
"human-pysi7_v3",
|
22 |
+
"humans-ziarm_v2",
|
23 |
+
"people-4evn7-fqlf8-d887c_v3",
|
24 |
+
"people-4evn7_v2",
|
25 |
+
"person-dataset-kzsop-vemv4-h1uoh-q5vtx_v2",
|
26 |
+
"tello-olz2y_v5"
|
27 |
+
],
|
28 |
+
"class_image_counts": {
|
29 |
+
"Person": 10865
|
30 |
+
}
|
31 |
},
|
32 |
{
|
33 |
+
"model_name": "face",
|
34 |
+
"display_name": "Face Detection Model",
|
35 |
+
"download_url": "https://raw.githubusercontent.com/wuhplaptop/face-11-n/main/face2.pt",
|
36 |
+
"architecture": "YOLOv11",
|
37 |
+
"training_epochs": 75,
|
38 |
+
"batch_size": 32,
|
39 |
+
"optimizer": "auto",
|
40 |
+
"learning_rate": 0.0005,
|
41 |
+
"data_augmentation_level": "Moderate",
|
42 |
+
"mAP_score": 0.99373,
|
43 |
+
"num_images": 15349,
|
44 |
+
"class_ids": {
|
45 |
+
"0": "Face"
|
46 |
+
},
|
47 |
+
"datasets_used": [
|
48 |
+
"face-detection-mik1i_v24",
|
49 |
+
"faces-bfigz_v3",
|
50 |
+
"head-qug6h_v2"
|
51 |
+
],
|
52 |
+
"class_image_counts": {
|
53 |
+
"Face": 15349
|
54 |
+
}
|
55 |
+
},
|
56 |
+
{
|
57 |
+
"model_name": "guns",
|
58 |
+
"display_name": "Guns and Knives Detection Model",
|
59 |
+
"download_url": "https://raw.githubusercontent.com/wuhplaptop/guns-100-11m/main/Guns-100-11m.pt",
|
60 |
+
"architecture": "YOLOv11",
|
61 |
+
"training_epochs": 100,
|
62 |
+
"batch_size": 32,
|
63 |
+
"optimizer": "auto",
|
64 |
+
"learning_rate": 0.0005,
|
65 |
+
"data_augmentation_level": "Moderate",
|
66 |
+
"mAP_score": 0.85476,
|
67 |
+
"num_images": 44219, // Sum of Gun and Knife image counts
|
68 |
+
"class_ids": {
|
69 |
+
"0": "Gun",
|
70 |
+
"1": "Knife"
|
71 |
+
},
|
72 |
+
"datasets_used": [
|
73 |
+
"gun-detection-xwosb_v8",
|
74 |
+
"gun-sqvhj_v1",
|
75 |
+
"people_with_guns_2-pvcnk_v9",
|
76 |
+
"police1_v1",
|
77 |
+
"soldier-wclp7_v1",
|
78 |
+
"test-e46au_v1",
|
79 |
+
"weapon-rl8c4_v3",
|
80 |
+
"weapondetection-fdzlo_v4",
|
81 |
+
"weapons-detection-x9fnq_v3",
|
82 |
+
"wepon_detection_v1"
|
83 |
+
],
|
84 |
+
"class_image_counts": {
|
85 |
+
"Gun": 44219,
|
86 |
+
"Knife": 4038
|
87 |
+
}
|
88 |
}
|
|
|
89 |
]
|