henry000 commited on
Commit
5b97cd6
·
1 Parent(s): 642e32f

🔧 [Fix] the datatype of args in model struct file

Browse files
Files changed (1) hide show
  1. config/model/v7-base.yaml +26 -22
config/model/v7-base.yaml CHANGED
@@ -1,5 +1,5 @@
1
- anchor:
2
- [1, 2, 3]
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
- - MP:
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
- - MP:
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
- - MP:
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
- - SPPCSPC:
119
- args: [512]
120
  - Conv:
121
  args: {out_channels: 256, kernel_size: 1}
122
- - Upsample:
123
- args: [None, 2, nearest]
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
- - Upsample:
149
- args: [None, 2, nearest]
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
- - MP:
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
- - MP:
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: [256, 3, 1]
230
  source: 75
231
  - RepConv:
232
- args: [512, 3, 1]
233
  source: 88
234
  - RepConv:
235
- args: [1024, 3, 1]
236
  source: 101
237
  - IDetect:
238
- args: [nc, anchors]
 
 
 
 
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]