Joseph Redmon
2016-09-01 8f1b4e0962857d402f9d017fcbf387ef0eceb7c4
cfg/yolo.cfg
@@ -1,27 +1,23 @@
[net]
batch=64
subdivisions=64
batch=1
subdivisions=1
height=448
width=448
channels=3
momentum=0.9
decay=0.0005
saturation=1.5
exposure=1.5
hue=.1
learning_rate=0.001
learning_rate=0.0005
policy=steps
steps=200,400,600,20000,30000
scales=2.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]
batch_normalize=1
filters=64
size=7
stride=2
@@ -33,6 +29,7 @@
stride=2
[convolutional]
batch_normalize=1
filters=192
size=3
stride=1
@@ -44,6 +41,7 @@
stride=2
[convolutional]
batch_normalize=1
filters=128
size=1
stride=1
@@ -51,6 +49,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=256
size=3
stride=1
@@ -58,6 +57,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=256
size=1
stride=1
@@ -65,6 +65,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=512
size=3
stride=1
@@ -76,6 +77,7 @@
stride=2
[convolutional]
batch_normalize=1
filters=256
size=1
stride=1
@@ -83,6 +85,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=512
size=3
stride=1
@@ -90,6 +93,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=256
size=1
stride=1
@@ -97,6 +101,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=512
size=3
stride=1
@@ -104,6 +109,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=256
size=1
stride=1
@@ -111,6 +117,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=512
size=3
stride=1
@@ -118,6 +125,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=256
size=1
stride=1
@@ -125,6 +133,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=512
size=3
stride=1
@@ -132,6 +141,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=512
size=1
stride=1
@@ -139,6 +149,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=1024
size=3
stride=1
@@ -150,6 +161,7 @@
stride=2
[convolutional]
batch_normalize=1
filters=512
size=1
stride=1
@@ -157,6 +169,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=1024
size=3
stride=1
@@ -164,6 +177,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=512
size=1
stride=1
@@ -171,6 +185,7 @@
activation=leaky
[convolutional]
batch_normalize=1
filters=1024
size=3
stride=1
@@ -180,6 +195,7 @@
#######
[convolutional]
batch_normalize=1
size=3
stride=1
pad=1
@@ -187,6 +203,7 @@
activation=leaky
[convolutional]
batch_normalize=1
size=3
stride=2
pad=1
@@ -194,6 +211,7 @@
activation=leaky
[convolutional]
batch_normalize=1
size=3
stride=1
pad=1
@@ -201,21 +219,25 @@
activation=leaky
[convolutional]
batch_normalize=1
size=3
stride=1
pad=1
filters=1024
activation=leaky
[connected]
output=4096
[local]
size=3
stride=1
pad=1
filters=256
activation=leaky
[dropout]
probability=.5
[connected]
output= 1470
output= 1715
activation=linear
[detection]
@@ -223,7 +245,7 @@
coords=4
rescore=1
side=7
num=2
num=3
softmax=0
sqrt=1
jitter=.2