From eaf033c0570308dfcd381ed61d274c7f5add7cfc Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 09 Nov 2015 21:27:02 +0000
Subject: [PATCH] Added tiny yolo model

---
 cfg/yolo-small.cfg |  182 +++++++++++++++++++++++++++------------------
 1 files changed, 108 insertions(+), 74 deletions(-)

diff --git a/cfg/yolo-small.cfg b/cfg/yolo-small.cfg
index e1be1ac..2a84485 100644
--- a/cfg/yolo-small.cfg
+++ b/cfg/yolo-small.cfg
@@ -4,202 +4,236 @@
 height=448
 width=448
 channels=3
-learning_rate=0.01
 momentum=0.9
 decay=0.0005
-seen = 0
+
+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
-exposure = 2
+saturation = 1.5
+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
-
-[convolutional]
-filters=256
-size=1
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=1024
-size=3
-stride=2
-pad=1
-activation=ramp
+activation=leaky
 
 [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
-activation=ramp
+filters=1024
+activation=leaky
 
 [connected]
-output=2048
-activation=ramp
+output=512
+activation=leaky
+
+[connected]
+output=4096
+activation=leaky
 
 [dropout]
 probability=.5
 
 [connected]
-output=1225
-activation=logistic
+output= 1470
+activation=linear
 
 [detection]
 classes=20
 coords=4
 rescore=1
-joint=1
-objectness = 0
-background=0
+side=7
+num=2
+softmax=0
+sqrt=1
+jitter=.2
+
+object_scale=1
+noobject_scale=.5
+class_scale=1
+coord_scale=5
+

--
Gitblit v1.10.0