mirror of https://github.com/AlexeyAB/darknet.git
Added tiny yolo model
This commit is contained in:
parent
ea4511006f
commit
eaf033c057
|
@ -1,6 +1,6 @@
|
||||||
[net]
|
[net]
|
||||||
batch=64
|
batch=64
|
||||||
subdivisions=2
|
subdivisions=64
|
||||||
height=448
|
height=448
|
||||||
width=448
|
width=448
|
||||||
channels=3
|
channels=3
|
||||||
|
|
|
@ -0,0 +1,140 @@
|
||||||
|
[net]
|
||||||
|
batch=64
|
||||||
|
subdivisions=64
|
||||||
|
height=448
|
||||||
|
width=448
|
||||||
|
channels=3
|
||||||
|
momentum=0.9
|
||||||
|
decay=0.0005
|
||||||
|
|
||||||
|
learning_rate=0.0001
|
||||||
|
policy=steps
|
||||||
|
steps=20,40,60,80,20000,30000
|
||||||
|
scales=5,5,2,2,.1,.1
|
||||||
|
max_batches = 40000
|
||||||
|
|
||||||
|
[crop]
|
||||||
|
crop_width=448
|
||||||
|
crop_height=448
|
||||||
|
flip=0
|
||||||
|
angle=0
|
||||||
|
saturation = 1.5
|
||||||
|
exposure = 1.5
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
filters=16
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[maxpool]
|
||||||
|
size=2
|
||||||
|
stride=2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
filters=32
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[maxpool]
|
||||||
|
size=2
|
||||||
|
stride=2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
filters=64
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[maxpool]
|
||||||
|
size=2
|
||||||
|
stride=2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
filters=128
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[maxpool]
|
||||||
|
size=2
|
||||||
|
stride=2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
filters=256
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[maxpool]
|
||||||
|
size=2
|
||||||
|
stride=2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
filters=512
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[maxpool]
|
||||||
|
size=2
|
||||||
|
stride=2
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
filters=1024
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
filters=1024
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[convolutional]
|
||||||
|
filters=1024
|
||||||
|
size=3
|
||||||
|
stride=1
|
||||||
|
pad=1
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[connected]
|
||||||
|
output=256
|
||||||
|
activation=linear
|
||||||
|
|
||||||
|
[connected]
|
||||||
|
output=4096
|
||||||
|
activation=leaky
|
||||||
|
|
||||||
|
[dropout]
|
||||||
|
probability=.5
|
||||||
|
|
||||||
|
[connected]
|
||||||
|
output= 1470
|
||||||
|
activation=linear
|
||||||
|
|
||||||
|
[detection]
|
||||||
|
classes=20
|
||||||
|
coords=4
|
||||||
|
rescore=1
|
||||||
|
side=7
|
||||||
|
num=2
|
||||||
|
softmax=0
|
||||||
|
sqrt=1
|
||||||
|
jitter=.2
|
||||||
|
|
||||||
|
object_scale=1
|
||||||
|
noobject_scale=.5
|
||||||
|
class_scale=1
|
||||||
|
coord_scale=5
|
||||||
|
|
Loading…
Reference in New Issue