From 9a01e6ccb7a74ff77e99060cf18acd6cfdb74b8e Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 11 Nov 2016 16:48:40 +0000
Subject: [PATCH] :fire: crush. crush. admit. :fire:

---
 cfg/yolo.cfg |   58 ++++++++++++++++++++++++++++++++++++++++------------------
 1 files changed, 40 insertions(+), 18 deletions(-)

diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg
index ee16726..c4f415c 100644
--- a/cfg/yolo.cfg
+++ b/cfg/yolo.cfg
@@ -1,27 +1,23 @@
 [net]
-batch=64
-subdivisions=2
+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=100,200,300,400,500,600,700,20000,30000
-scales=2,2,1.25,1.25,1.25,1.25,1.03,.1,.1
+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,29 +219,33 @@
 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
 
-[region]
+[detection]
 classes=20
 coords=4
 rescore=1
 side=7
-num=2
+num=3
 softmax=0
 sqrt=1
 jitter=.2

--
Gitblit v1.10.0