🔧 [Fix] the datatype of args in model struct file
Browse files- config/model/v7-base.yaml +26 -22
config/model/v7-base.yaml
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
-
|
2 |
-
|
3 |
model:
|
4 |
backbone:
|
5 |
- Conv:
|
@@ -27,8 +27,8 @@ model:
|
|
27 |
source: [-1, -3, -5, -6]
|
28 |
- Conv:
|
29 |
args: {out_channels: 256, kernel_size: 1}
|
30 |
-
-
|
31 |
-
args:
|
32 |
- Conv:
|
33 |
args: {out_channels: 128, kernel_size: 1}
|
34 |
- Conv:
|
@@ -56,8 +56,8 @@ model:
|
|
56 |
tags: 8x
|
57 |
- Conv:
|
58 |
args: {out_channels: 512, kernel_size: 1}
|
59 |
-
-
|
60 |
-
args:
|
61 |
- Conv:
|
62 |
args: {out_channels: 256, kernel_size: 1}
|
63 |
- Conv:
|
@@ -85,8 +85,8 @@ model:
|
|
85 |
- Conv:
|
86 |
args: {out_channels: 1024, kernel_size: 1}
|
87 |
tags: 16x
|
88 |
-
-
|
89 |
-
args:
|
90 |
- Conv:
|
91 |
args: {out_channels: 512, kernel_size: 1}
|
92 |
- Conv:
|
@@ -115,12 +115,12 @@ model:
|
|
115 |
args: {out_channels: 1024, kernel_size: 1}
|
116 |
tags: 32x
|
117 |
head:
|
118 |
-
-
|
119 |
-
args:
|
120 |
- Conv:
|
121 |
args: {out_channels: 256, kernel_size: 1}
|
122 |
-
-
|
123 |
-
args:
|
124 |
- Conv:
|
125 |
args: {out_channels: 256, kernel_size: 1}
|
126 |
source: 16x
|
@@ -145,8 +145,8 @@ model:
|
|
145 |
args: {out_channels: 256, kernel_size: 1}
|
146 |
- Conv:
|
147 |
args: {out_channels: 128, kernel_size: 1}
|
148 |
-
-
|
149 |
-
args:
|
150 |
- Conv:
|
151 |
args: {out_channels: 128, kernel_size: 1}
|
152 |
source: 8x
|
@@ -169,8 +169,8 @@ model:
|
|
169 |
source: [-1, -2, -3, -4, -5, -6]
|
170 |
- Conv:
|
171 |
args: {out_channels: 128, kernel_size: 1}
|
172 |
-
-
|
173 |
-
args:
|
174 |
- Conv:
|
175 |
args: {out_channels: 128, kernel_size: 1}
|
176 |
- Conv:
|
@@ -197,8 +197,8 @@ model:
|
|
197 |
source: [-1, -2, -3, -4, -5, -6]
|
198 |
- Conv:
|
199 |
args: {out_channels: 256, kernel_size: 1}
|
200 |
-
-
|
201 |
-
args:
|
202 |
- Conv:
|
203 |
args: {out_channels: 256, kernel_size: 1}
|
204 |
- Conv:
|
@@ -226,14 +226,18 @@ model:
|
|
226 |
- Conv:
|
227 |
args: {out_channels: 512, kernel_size: 1}
|
228 |
- RepConv:
|
229 |
-
args:
|
230 |
source: 75
|
231 |
- RepConv:
|
232 |
-
args:
|
233 |
source: 88
|
234 |
- RepConv:
|
235 |
-
args:
|
236 |
source: 101
|
237 |
- IDetect:
|
238 |
-
args:
|
|
|
|
|
|
|
|
|
239 |
source: [102, 103, 104]
|
|
|
1 |
+
nc: 80
|
2 |
+
|
3 |
model:
|
4 |
backbone:
|
5 |
- Conv:
|
|
|
27 |
source: [-1, -3, -5, -6]
|
28 |
- Conv:
|
29 |
args: {out_channels: 256, kernel_size: 1}
|
30 |
+
- MaxPool:
|
31 |
+
args: {}
|
32 |
- Conv:
|
33 |
args: {out_channels: 128, kernel_size: 1}
|
34 |
- Conv:
|
|
|
56 |
tags: 8x
|
57 |
- Conv:
|
58 |
args: {out_channels: 512, kernel_size: 1}
|
59 |
+
- MaxPool:
|
60 |
+
args: {}
|
61 |
- Conv:
|
62 |
args: {out_channels: 256, kernel_size: 1}
|
63 |
- Conv:
|
|
|
85 |
- Conv:
|
86 |
args: {out_channels: 1024, kernel_size: 1}
|
87 |
tags: 16x
|
88 |
+
- MaxPool:
|
89 |
+
args: {}
|
90 |
- Conv:
|
91 |
args: {out_channels: 512, kernel_size: 1}
|
92 |
- Conv:
|
|
|
115 |
args: {out_channels: 1024, kernel_size: 1}
|
116 |
tags: 32x
|
117 |
head:
|
118 |
+
- SPPCSPConv:
|
119 |
+
args: {out_channels: 512}
|
120 |
- Conv:
|
121 |
args: {out_channels: 256, kernel_size: 1}
|
122 |
+
- UpSample:
|
123 |
+
args: {scale_factor: 2}
|
124 |
- Conv:
|
125 |
args: {out_channels: 256, kernel_size: 1}
|
126 |
source: 16x
|
|
|
145 |
args: {out_channels: 256, kernel_size: 1}
|
146 |
- Conv:
|
147 |
args: {out_channels: 128, kernel_size: 1}
|
148 |
+
- UpSample:
|
149 |
+
args: {scale_factor: 2}
|
150 |
- Conv:
|
151 |
args: {out_channels: 128, kernel_size: 1}
|
152 |
source: 8x
|
|
|
169 |
source: [-1, -2, -3, -4, -5, -6]
|
170 |
- Conv:
|
171 |
args: {out_channels: 128, kernel_size: 1}
|
172 |
+
- MaxPool:
|
173 |
+
args: {}
|
174 |
- Conv:
|
175 |
args: {out_channels: 128, kernel_size: 1}
|
176 |
- Conv:
|
|
|
197 |
source: [-1, -2, -3, -4, -5, -6]
|
198 |
- Conv:
|
199 |
args: {out_channels: 256, kernel_size: 1}
|
200 |
+
- MaxPool:
|
201 |
+
args: {}
|
202 |
- Conv:
|
203 |
args: {out_channels: 256, kernel_size: 1}
|
204 |
- Conv:
|
|
|
226 |
- Conv:
|
227 |
args: {out_channels: 512, kernel_size: 1}
|
228 |
- RepConv:
|
229 |
+
args: {out_channels: 256}
|
230 |
source: 75
|
231 |
- RepConv:
|
232 |
+
args: {out_channels: 512}
|
233 |
source: 88
|
234 |
- RepConv:
|
235 |
+
args: {out_channels: 1024}
|
236 |
source: 101
|
237 |
- IDetect:
|
238 |
+
args:
|
239 |
+
anchors:
|
240 |
+
- [12,16, 19,36, 40,28] # P3/8
|
241 |
+
- [36,75, 76,55, 72,146] # P4/16
|
242 |
+
- [142,110, 192,243, 459,401] # P5/32
|
243 |
source: [102, 103, 104]
|