changed yolo-small

cfg/yolo-small.cfg

@@ -1,204 +1,239 @@
 [net]
 batch=64
-subdivisions=64
+subdivisions=2
 height=448
 width=448
 channels=3
-learning_rate=0.01
 momentum=0.9
 decay=0.0005
 
+learning_rate=0.001
+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 = 2
+saturation = 1.5
-exposure = 2
+exposure = 1.5
 
 [convolutional]
 filters=64
 size=7
 stride=2
 pad=1
-activation=ramp
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
 
 [convolutional]
 filters=192
 size=3
-stride=2
+stride=1
 pad=1
-activation=ramp
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
 
 [convolutional]
 filters=128
 size=1
 stride=1
 pad=1
-activation=ramp
+activation=leaky
 
 [convolutional]
 filters=256
 size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+filters=256
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+filters=512
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[maxpool]
+size=2
 stride=2
-pad=1
-activation=ramp
-
-[convolutional]
-filters=128
-size=1
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=256
-size=3
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=128
-size=1
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=512
-size=3
-stride=2
-pad=1
-activation=ramp
 
 [convolutional]
 filters=256
 size=1
 stride=1
 pad=1
-activation=ramp
+activation=leaky
 
 [convolutional]
 filters=512
 size=3
 stride=1
 pad=1
-activation=ramp
+activation=leaky
 
 [convolutional]
 filters=256
 size=1
 stride=1
 pad=1
-activation=ramp
+activation=leaky
 
 [convolutional]
 filters=512
 size=3
 stride=1
 pad=1
-activation=ramp
+activation=leaky
 
 [convolutional]
 filters=256
 size=1
 stride=1
 pad=1
-activation=ramp
+activation=leaky
 
 [convolutional]
 filters=512
 size=3
 stride=1
 pad=1
-activation=ramp
+activation=leaky
 
 [convolutional]
 filters=256
 size=1
 stride=1
 pad=1
-activation=ramp
+activation=leaky
 
 [convolutional]
 filters=512
 size=3
 stride=1
 pad=1
-activation=ramp
+activation=leaky
-
-[convolutional]
-filters=256
-size=1
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=1024
-size=3
-stride=2
-pad=1
-activation=ramp
 
 [convolutional]
 filters=512
 size=1
 stride=1
 pad=1
-activation=ramp
+activation=leaky
 
 [convolutional]
 filters=1024
 size=3
 stride=1
 pad=1
-activation=ramp
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+filters=512
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+filters=1024
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+filters=512
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+filters=1024
+size=3
+stride=1
+pad=1
+activation=leaky
+
+#######
 
 [convolutional]
 size=3
 stride=1
 pad=1
 filters=1024
-activation=ramp
+activation=leaky
 
 [convolutional]
 size=3
 stride=2
 pad=1
 filters=1024
-activation=ramp
+activation=leaky
 
 [convolutional]
 size=3
 stride=1
 pad=1
 filters=1024
-activation=ramp
+activation=leaky
 
 [convolutional]
 size=3
 stride=1
 pad=1
-filters=512
+filters=1024
-activation=ramp
+activation=leaky
 
 [connected]
-output=2048
+output=512
-activation=ramp
+activation=leaky
+
+[connected]
+output=4096
+activation=leaky
 
 [dropout]
 probability=.5
 
 [connected]
-output=1225
+output= 1470
-activation=logistic
+activation=linear
 
 [detection]
 classes=20
 coords=4
 rescore=1
-joint=0
+side=7
-objectness=1
+num=2
+softmax=0
+sqrt=1
+jitter=.2
+
+object_scale=1
+noobject_scale=.5
+class_scale=1
+coord_scale=5