Skip to content

Commit 11d7276

Browse files
FlovaFlova
authored
Merge pull request #85 from bit-bots/feature/light_decoder
Add first version of the efficient decoder
2 parents 718ab5f + 78dd2a2 commit 11d7276

File tree

4 files changed

+1615
-1033
lines changed

4 files changed

+1615
-1033
lines changed

‎config/yoeo_light_decoder.cfg

Lines changed: 364 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,364 @@
1+
[net]
2+
# Testing
3+
#batch=1
4+
#subdivisions=1
5+
# Training
6+
batch=64
7+
subdivisions=4
8+
width=416
9+
height=416
10+
channels=3
11+
momentum=0.9
12+
decay=0.0005
13+
angle=0
14+
saturation = 1.5
15+
exposure = 1.5
16+
hue=.1
17+
18+
learning_rate=0.0001
19+
burn_in=100
20+
max_batches = 4000
21+
policy=steps
22+
steps=50000,60000
23+
scales=.1,.1
24+
25+
[convolutional]
26+
batch_normalize=1
27+
filters=32
28+
size=3
29+
stride=2
30+
pad=1
31+
activation=leaky
32+
33+
[convolutional]
34+
batch_normalize=1
35+
filters=64
36+
size=3
37+
stride=2
38+
pad=1
39+
activation=leaky
40+
41+
[convolutional]
42+
batch_normalize=1
43+
filters=64
44+
size=3
45+
stride=1
46+
pad=1
47+
activation=leaky
48+
49+
[route]
50+
layers=-1
51+
groups=2
52+
group_id=1
53+
54+
[convolutional]
55+
batch_normalize=1
56+
filters=32
57+
size=3
58+
stride=1
59+
pad=1
60+
activation=leaky
61+
62+
[convolutional]
63+
batch_normalize=1
64+
filters=32
65+
size=3
66+
stride=1
67+
pad=1
68+
activation=leaky
69+
70+
[route]
71+
layers = -1,-2
72+
73+
[convolutional]
74+
batch_normalize=1
75+
filters=64
76+
size=1
77+
stride=1
78+
pad=1
79+
activation=leaky
80+
81+
[route]
82+
layers = -6,-1
83+
84+
[maxpool]
85+
size=2
86+
stride=2
87+
88+
[convolutional]
89+
batch_normalize=1
90+
filters=128
91+
size=3
92+
stride=1
93+
pad=1
94+
activation=leaky
95+
96+
[route]
97+
layers=-1
98+
groups=2
99+
group_id=1
100+
101+
[convolutional]
102+
batch_normalize=1
103+
filters=64
104+
size=3
105+
stride=1
106+
pad=1
107+
activation=leaky
108+
109+
[convolutional]
110+
batch_normalize=1
111+
filters=64
112+
size=3
113+
stride=1
114+
pad=1
115+
activation=leaky
116+
117+
[route]
118+
layers = -1,-2
119+
120+
[convolutional]
121+
batch_normalize=1
122+
filters=128
123+
size=1
124+
stride=1
125+
pad=1
126+
activation=leaky
127+
128+
[route]
129+
layers = -6,-1
130+
131+
[maxpool]
132+
size=2
133+
stride=2
134+
135+
[convolutional]
136+
batch_normalize=1
137+
filters=256
138+
size=3
139+
stride=1
140+
pad=1
141+
activation=leaky
142+
143+
[route]
144+
layers=-1
145+
groups=2
146+
group_id=1
147+
148+
[convolutional]
149+
batch_normalize=1
150+
filters=128
151+
size=3
152+
stride=1
153+
pad=1
154+
activation=leaky
155+
156+
[convolutional]
157+
batch_normalize=1
158+
filters=128
159+
size=3
160+
stride=1
161+
pad=1
162+
activation=leaky
163+
164+
[route]
165+
layers = -1,-2
166+
167+
[convolutional]
168+
batch_normalize=1
169+
filters=256
170+
size=1
171+
stride=1
172+
pad=1
173+
activation=leaky
174+
175+
[route]
176+
layers = -6,-1
177+
178+
[maxpool]
179+
size=2
180+
stride=2
181+
182+
[convolutional]
183+
batch_normalize=1
184+
filters=512
185+
size=3
186+
stride=1
187+
pad=1
188+
activation=leaky
189+
190+
###########
191+
192+
[convolutional]
193+
batch_normalize=1
194+
filters=256
195+
size=1
196+
stride=1
197+
pad=1
198+
activation=leaky
199+
200+
[convolutional]
201+
batch_normalize=1
202+
filters=512
203+
size=3
204+
stride=1
205+
pad=1
206+
activation=leaky
207+
208+
[convolutional]
209+
size=1
210+
stride=1
211+
pad=1
212+
filters=8
213+
activation=linear
214+
215+
[yolo]
216+
mask = 0
217+
anchors = 100, 100
218+
classes=3
219+
num=1
220+
jitter=.3
221+
ignore_thresh = .7
222+
truth_thresh = 1
223+
random=1
224+
225+
[route]
226+
layers = -4
227+
228+
[convolutional]
229+
batch_normalize=1
230+
filters=128
231+
size=1
232+
stride=1
233+
pad=1
234+
activation=leaky
235+
236+
[upsample]
237+
stride=2
238+
239+
[route]
240+
layers = -1, 24
241+
242+
[convolutional]
243+
batch_normalize=1
244+
filters=256
245+
size=3
246+
stride=1
247+
pad=1
248+
activation=leaky
249+
250+
[convolutional]
251+
size=1
252+
stride=1
253+
pad=1
254+
filters=8
255+
activation=linear
256+
257+
[yolo]
258+
mask = 0
259+
anchors = 100,100
260+
classes=3
261+
num=1
262+
jitter=.3
263+
ignore_thresh = .7
264+
truth_thresh = 1
265+
random=1
266+
267+
[route]
268+
layers = 18
269+
270+
[convolutional]
271+
batch_normalize=1
272+
filters=128
273+
size=1
274+
stride=1
275+
pad=1
276+
activation=leaky
277+
278+
[upsample]
279+
stride=2
280+
281+
[route]
282+
layers = -1, 10
283+
284+
[convolutional]
285+
batch_normalize=1
286+
filters=64
287+
size=1
288+
stride=1
289+
pad=1
290+
activation=leaky
291+
292+
[convolutional]
293+
batch_normalize=1
294+
filters=64
295+
size=3
296+
stride=1
297+
pad=1
298+
activation=leaky
299+
300+
[upsample]
301+
stride=2
302+
303+
[route]
304+
layers = -1, 2
305+
306+
[convolutional]
307+
batch_normalize=1
308+
filters=32
309+
size=1
310+
stride=1
311+
pad=1
312+
activation=leaky
313+
314+
[convolutional]
315+
batch_normalize=1
316+
filters=32
317+
size=3
318+
stride=1
319+
pad=1
320+
activation=leaky
321+
322+
[upsample]
323+
stride=2
324+
325+
[route]
326+
layers = -1, 0
327+
328+
[convolutional]
329+
batch_normalize=1
330+
filters=16
331+
size=1
332+
stride=1
333+
pad=1
334+
activation=leaky
335+
336+
[convolutional]
337+
batch_normalize=1
338+
filters=16
339+
size=3
340+
stride=1
341+
pad=1
342+
activation=leaky
343+
344+
[upsample]
345+
stride=2
346+
347+
[convolutional]
348+
batch_normalize=1
349+
filters=16
350+
size=3
351+
stride=1
352+
pad=1
353+
activation=leaky
354+
355+
[convolutional]
356+
batch_normalize=1
357+
filters=3
358+
size=1
359+
stride=1
360+
pad=1
361+
activation=linear
362+
363+
[seg]
364+
classes=3

0 commit comments

Comments
 (0)