From 845ab7579685b6702c92c1088ec11e71bde51f3c Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 05 Aug 2016 22:27:07 +0000
Subject: [PATCH] some more stuff

---
 cfg/extraction.cfg |   50 +++++++++++++++++++++++++++++++++-----------------
 1 files changed, 33 insertions(+), 17 deletions(-)

diff --git a/cfg/extraction.cfg b/cfg/extraction.cfg
index f9668a5..94e1067 100644
--- a/cfg/extraction.cfg
+++ b/cfg/extraction.cfg
@@ -1,26 +1,20 @@
 [net]
 batch=128
 subdivisions=1
-height=256
-width=256
+height=224
+width=224
+max_crop=320
 channels=3
 momentum=0.9
 decay=0.0005
 
-learning_rate=0.5
+learning_rate=0.1
 policy=poly
-power=6
-max_batches=500000
-
-[crop]
-crop_height=224
-crop_width=224
-flip=1
-saturation=1
-exposure=1
-angle=0
+power=4
+max_batches=1600000
 
 [convolutional]
+batch_normalize=1
 filters=64
 size=7
 stride=2
@@ -32,6 +26,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=192
 size=3
 stride=1
@@ -43,6 +38,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=128
 size=1
 stride=1
@@ -50,6 +46,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=3
 stride=1
@@ -57,6 +54,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -64,6 +62,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -75,6 +74,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -82,6 +82,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -89,6 +90,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -96,6 +98,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -103,6 +106,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -110,6 +114,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -117,6 +122,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -124,6 +130,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -131,6 +138,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -138,6 +146,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -149,6 +158,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -156,6 +166,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -163,6 +174,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -170,18 +182,22 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
 pad=1
 activation=leaky
 
+[convolutional]
+filters=1000
+size=1
+stride=1
+pad=1
+activation=leaky
+
 [avgpool]
 
-[connected]
-output=1000
-activation=leaky
-
 [softmax]
 groups=1
 

--
Gitblit v1.10.0