From 6b38dcdce07b0c082334e7d9e10fe26bc440a347 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 08 Sep 2016 05:27:56 +0000
Subject: [PATCH] adding new tiny-yolo

---
 cfg/tiny-yolo.cfg |   50 ++++++++++++++++++--------------------------------
 1 files changed, 18 insertions(+), 32 deletions(-)

diff --git a/cfg/yolo-tiny.cfg b/cfg/tiny-yolo.cfg
similarity index 72%
rename from cfg/yolo-tiny.cfg
rename to cfg/tiny-yolo.cfg
index c72ed9b..ac4b346 100644
--- a/cfg/yolo-tiny.cfg
+++ b/cfg/tiny-yolo.cfg
@@ -1,27 +1,24 @@
 [net]
 batch=64
-subdivisions=64
+subdivisions=2
 height=448
 width=448
 channels=3
 momentum=0.9
 decay=0.0005
 
-learning_rate=0.0001
+saturation=.75
+exposure=.75
+hue = .1
+
+learning_rate=0.0005
 policy=steps
-steps=20,40,60,80,20000,30000
-scales=5,5,2,2,.1,.1
+steps=200,400,600,800,20000,30000
+scales=2.5,2,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=16
 size=3
 stride=1
@@ -33,6 +30,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=32
 size=3
 stride=1
@@ -44,6 +42,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=64
 size=3
 stride=1
@@ -55,6 +54,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=128
 size=3
 stride=1
@@ -66,6 +66,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=3
 stride=1
@@ -77,6 +78,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -88,37 +90,21 @@
 stride=2
 
 [convolutional]
-filters=1024
+batch_normalize=1
 size=3
 stride=1
 pad=1
+filters=1024
 activation=leaky
 
 [convolutional]
-filters=1024
+batch_normalize=1
 size=3
 stride=1
 pad=1
+filters=256
 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

--
Gitblit v1.10.0