wuhp commited on
Commit
753e2c0
·
verified ·
1 Parent(s): d957ce9

Update models_info.json

Browse files
Files changed (1) hide show
  1. models_info.json +83 -7
models_info.json CHANGED
@@ -1,13 +1,89 @@
1
  [
2
  {
3
- "model_name": "model1",
4
- "mAP_score": 0.85,
5
- "num_images": 1000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  },
7
  {
8
- "model_name": "model2",
9
- "mAP_score": 0.90,
10
- "num_images": 1500
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
  ]