From 8f1b4e0962857d402f9d017fcbf387ef0eceb7c4 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 01 Sep 2016 23:48:41 +0000
Subject: [PATCH] updates and things

---
 cfg/yolo.cfg |   52 +++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 37 insertions(+), 15 deletions(-)

diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg
index 8664831..c4f415c 100644
--- a/cfg/yolo.cfg
+++ b/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

--
Gitblit v1.10.0