Hugging Face's logo

Hsueh1001
/
28-yolo

Object Detection
Transformers
PyTorch
yolos
Inference Endpoints
Model card Files Community
Hsueh1001 commited on
Commit
034959e
•
1 Parent(s): b25638a

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +132 -4
config.json CHANGED
@@ -4,16 +4,144 @@
4
  ],
5
  "model_type": "yolos",
6
  "input_size": [640, 640],
 
 
 
7
  "anchors": [
8
- [10, 13],
9
- [16, 30],
10
- [33, 23]
11
  ],
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  "classes": ["0", "1"],
13
  "output_layers": [
14
  {"layer_size": [52, 52], "anchors": 3},
15
  {"layer_size": [26, 26], "anchors": 3},
16
  {"layer_size": [13, 13], "anchors": 3}
17
  ],
18
- "nms_threshold": 0.5
19
  }
 
 
4
  ],
5
  "model_type": "yolos",
6
  "input_size": [640, 640],
7
+ "nc": 2,
8
+ "depth_multiple": 1.0,
9
+ "width_multiple": 1.0,
10
  "anchors": [
11
+ [12, 16, 19, 36, 40, 28],
12
+ [36, 75, 76, 55, 72, 146],
13
+ [142, 110, 192, 243, 459, 401]
14
  ],
15
+ "backbone":[
16
+ [-1, 1, Conv, [32, 3, 1]],
17
+
18
+ [-1, 1, Conv, [64, 3, 2]],
19
+ [-1, 1, Conv, [64, 3, 1]],
20
+
21
+ [-1, 1, Conv, [128, 3, 2]],
22
+ [-1, 1, Conv, [64, 1, 1]],
23
+ [-2, 1, Conv, [64, 1, 1]],
24
+ [-1, 1, Conv, [64, 3, 1]],
25
+ [-1, 1, Conv, [64, 3, 1]],
26
+ [-1, 1, Conv, [64, 3, 1]],
27
+ [-1, 1, Conv, [64, 3, 1]],
28
+ [[-1, -3, -5, -6], 1, Concat, [1]],
29
+ [-1, 1, Conv, [256, 1, 1]],
30
+
31
+ [-1, 1, MP, []],
32
+ [-1, 1, Conv, [128, 1, 1]],
33
+ [-3, 1, Conv, [128, 1, 1]],
34
+ [-1, 1, Conv, [128, 3, 2]],
35
+ [[-1, -3], 1, Concat, [1]],
36
+ [-1, 1, Conv, [128, 1, 1]],
37
+ [-2, 1, Conv, [128, 1, 1]],
38
+ [-1, 1, Conv, [128, 3, 1]],
39
+ [-1, 1, Conv, [128, 3, 1]],
40
+ [-1, 1, Conv, [128, 3, 1]],
41
+ [-1, 1, Conv, [128, 3, 1]],
42
+ [[-1, -3, -5, -6], 1, Concat, [1]],
43
+ [-1, 1, Conv, [512, 1, 1]],
44
+
45
+ [-1, 1, MP, []],
46
+ [-1, 1, Conv, [256, 1, 1]],
47
+ [-3, 1, Conv, [256, 1, 1]],
48
+ [-1, 1, Conv, [256, 3, 2]],
49
+ [[-1, -3], 1, Concat, [1]],
50
+ [-1, 1, Conv, [256, 1, 1]],
51
+ [-2, 1, Conv, [256, 1, 1]],
52
+ [-1, 1, Conv, [256, 3, 1]],
53
+ [-1, 1, Conv, [256, 3, 1]],
54
+ [-1, 1, Conv, [256, 3, 1]],
55
+ [-1, 1, Conv, [256, 3, 1]],
56
+ [[-1, -3, -5, -6], 1, Concat, [1]],
57
+ [-1, 1, Conv, [1024, 1, 1]],
58
+
59
+ [-1, 1, MP, []],
60
+ [-1, 1, Conv, [512, 1, 1]],
61
+ [-3, 1, Conv, [512, 1, 1]],
62
+ [-1, 1, Conv, [512, 3, 2]],
63
+ [[-1, -3], 1, Concat, [1]],
64
+ [-1, 1, Conv, [256, 1, 1]],
65
+ [-2, 1, Conv, [256, 1, 1]],
66
+ [-1, 1, Conv, [256, 3, 1]],
67
+ [-1, 1, Conv, [256, 3, 1]],
68
+ [-1, 1, Conv, [256, 3, 1]],
69
+ [-1, 1, Conv, [256, 3, 1]],
70
+ [[-1, -3, -5, -6], 1, Concat, [1]],
71
+ [-1, 1, Conv, [1024, 1, 1]],
72
+ ],
73
+ "head":[
74
+ [-1, 1, SPPCSPC, [512]],
75
+
76
+ [-1, 1, Conv, [256, 1, 1]],
77
+ [-1, 1, nn.Upsample, [None, 2, 'nearest']],
78
+ [37, 1, Conv, [256, 1, 1]],
79
+ [[-1, -2], 1, Concat, [1]],
80
+
81
+ [-1, 1, Conv, [256, 1, 1]],
82
+ [-2, 1, Conv, [256, 1, 1]],
83
+ [-1, 1, Conv, [128, 3, 1]],
84
+ [-1, 1, Conv, [128, 3, 1]],
85
+ [-1, 1, Conv, [128, 3, 1]],
86
+ [-1, 1, Conv, [128, 3, 1]],
87
+ [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]],
88
+ [-1, 1, Conv, [256, 1, 1]],
89
+
90
+ [-1, 1, Conv, [128, 1, 1]],
91
+ [-1, 1, nn.Upsample, [None, 2, 'nearest']],
92
+ [24, 1, Conv, [128, 1, 1]],
93
+ [[-1, -2], 1, Concat, [1]],
94
+
95
+ [-1, 1, Conv, [128, 1, 1]],
96
+ [-2, 1, Conv, [128, 1, 1]],
97
+ [-1, 1, Conv, [64, 3, 1]],
98
+ [-1, 1, Conv, [64, 3, 1]],
99
+ [-1, 1, Conv, [64, 3, 1]],
100
+ [-1, 1, Conv, [64, 3, 1]],
101
+ [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]],
102
+ [-1, 1, Conv, [128, 1, 1]],
103
+
104
+ [-1, 1, MP, []],
105
+ [-1, 1, Conv, [128, 1, 1]],
106
+ [-3, 1, Conv, [128, 1, 1]],
107
+ [-1, 1, Conv, [128, 3, 2]],
108
+ [[-1, -3, 63], 1, Concat, [1]],
109
+
110
+ [-1, 1, Conv, [256, 1, 1]],
111
+ [-2, 1, Conv, [256, 1, 1]],
112
+ [-1, 1, Conv, [128, 3, 1]],
113
+ [-1, 1, Conv, [128, 3, 1]],
114
+ [-1, 1, Conv, [128, 3, 1]],
115
+ [-1, 1, Conv, [128, 3, 1]],
116
+ [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]],
117
+ [-1, 1, Conv, [256, 1, 1]],
118
+
119
+ [-1, 1, MP, []],
120
+ [-1, 1, Conv, [256, 1, 1]],
121
+ [-3, 1, Conv, [256, 1, 1]],
122
+ [-1, 1, Conv, [256, 3, 2]],
123
+ [[-1, -3, 51], 1, Concat, [1]],
124
+
125
+ [-1, 1, Conv, [512, 1, 1]],
126
+ [-2, 1, Conv, [512, 1, 1]],
127
+ [-1, 1, Conv, [256, 3, 1]],
128
+ [-1, 1, Conv, [256, 3, 1]],
129
+ [-1, 1, Conv, [256, 3, 1]],
130
+ [-1, 1, Conv, [256, 3, 1]],
131
+ [[-1, -2, -3, -4, -5, -6], 1, Concat, [1]],
132
+ [-1, 1, Conv, [512, 1, 1]],
133
+
134
+ [75, 1, RepConv, [256, 3, 1]],
135
+ [88, 1, RepConv, [512, 3, 1]],
136
+ [101, 1, RepConv, [1024, 3, 1]],
137
+
138
+ [[102,103,104], 1, IDetect, [nc, anchors]],
139
  "classes": ["0", "1"],
140
  "output_layers": [
141
  {"layer_size": [52, 52], "anchors": 3},
142
  {"layer_size": [26, 26], "anchors": 3},
143
  {"layer_size": [13, 13], "anchors": 3}
144
  ],
145
+ "nms_threshold": 0.1
146
  }
147
+