From aebe937710ced03d03f73ab23f410f29685655c1 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 11 Aug 2016 18:54:24 +0000
Subject: [PATCH] what do you even write here?

---
 cfg/yolo.cfg |   53 ++++++++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 46 insertions(+), 7 deletions(-)

diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg
index 1f69856..6876eff 100644
--- a/cfg/yolo.cfg
+++ b/cfg/yolo.cfg
@@ -1,19 +1,29 @@
 [net]
-batch=1
-subdivisions=1
+batch=64
+subdivisions=2
 height=448
 width=448
 channels=3
 momentum=0.9
 decay=0.0005
 
-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
+noadjust=1
+
 [convolutional]
+batch_normalize=1
 filters=64
 size=7
 stride=2
@@ -25,6 +35,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=192
 size=3
 stride=1
@@ -36,6 +47,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=128
 size=1
 stride=1
@@ -43,6 +55,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=3
 stride=1
@@ -50,6 +63,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -57,6 +71,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -68,6 +83,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -75,6 +91,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -82,6 +99,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -89,6 +107,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -96,6 +115,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -103,6 +123,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -110,6 +131,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -117,6 +139,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -124,6 +147,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -131,6 +155,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -142,6 +167,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -149,6 +175,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -156,6 +183,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -163,6 +191,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -172,6 +201,7 @@
 #######
 
 [convolutional]
+batch_normalize=1
 size=3
 stride=1
 pad=1
@@ -179,6 +209,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 size=3
 stride=2
 pad=1
@@ -186,6 +217,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 size=3
 stride=1
 pad=1
@@ -193,18 +225,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]
@@ -212,7 +251,7 @@
 coords=4
 rescore=1
 side=7
-num=2
+num=3
 softmax=0
 sqrt=1
 jitter=.2

--
Gitblit v1.10.0