From 73a2c52df098ecc800255d9cf8ee5d565ef532ab Mon Sep 17 00:00:00 2001
From: Alexey <AlexeyAB@users.noreply.github.com>
Date: Fri, 01 Jun 2018 11:51:36 +0000
Subject: [PATCH] Merge pull request #946 from jinyu121/master

---
 cfg/darknet.cfg |   55 +++++++++++++++++++++++++++++--------------------------
 1 files changed, 29 insertions(+), 26 deletions(-)

diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg
index f52ff3f..60b939a 100644
--- a/cfg/darknet.cfg
+++ b/cfg/darknet.cfg
@@ -1,25 +1,20 @@
 [net]
 batch=128
 subdivisions=1
-height=256
-width=256
+height=224
+width=224
 channels=3
 momentum=0.9
 decay=0.0005
-learning_rate=0.01
-policy=poly
-power=.5
-max_batches=600000
+max_crop=320
 
-[crop]
-crop_height=224
-crop_width=224
-flip=1
-angle=0
-saturation=1
-exposure=1
+learning_rate=0.1
+policy=poly
+power=4
+max_batches=1600000
 
 [convolutional]
+batch_normalize=1
 filters=16
 size=3
 stride=1
@@ -27,10 +22,11 @@
 activation=leaky
 
 [maxpool]
-size=3
+size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=32
 size=3
 stride=1
@@ -38,10 +34,11 @@
 activation=leaky
 
 [maxpool]
-size=3
+size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=64
 size=3
 stride=1
@@ -49,10 +46,11 @@
 activation=leaky
 
 [maxpool]
-size=3
+size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=128
 size=3
 stride=1
@@ -60,10 +58,11 @@
 activation=leaky
 
 [maxpool]
-size=3
+size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=3
 stride=1
@@ -71,10 +70,11 @@
 activation=leaky
 
 [maxpool]
-size=3
+size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -82,26 +82,29 @@
 activation=leaky
 
 [maxpool]
-size=3
+size=2
 stride=2
+padding=1
 
 [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]
 
-[dropout]
-probability=.5
-
-[connected]
-output=1000
-activation=linear
-
 [softmax]
+groups=1
 
 [cost]
 type=sse

--
Gitblit v1.10.0