From 352ae7e65b6a74bcd768aa88b866a44c713284c8 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 26 Oct 2016 15:35:44 +0000
Subject: [PATCH] ADAM

---
 cfg/yolo.cfg |   43 ++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 38 insertions(+), 5 deletions(-)

diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg
index 1f69856..c4f415c 100644
--- a/cfg/yolo.cfg
+++ b/cfg/yolo.cfg
@@ -6,14 +6,18 @@
 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
 
 [convolutional]
+batch_normalize=1
 filters=64
 size=7
 stride=2
@@ -25,6 +29,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=192
 size=3
 stride=1
@@ -36,6 +41,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=128
 size=1
 stride=1
@@ -43,6 +49,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=3
 stride=1
@@ -50,6 +57,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -57,6 +65,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -68,6 +77,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -75,6 +85,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -82,6 +93,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -89,6 +101,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -96,6 +109,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -103,6 +117,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -110,6 +125,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -117,6 +133,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -124,6 +141,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -131,6 +149,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -142,6 +161,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -149,6 +169,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -156,6 +177,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -163,6 +185,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -172,6 +195,7 @@
 #######
 
 [convolutional]
+batch_normalize=1
 size=3
 stride=1
 pad=1
@@ -179,6 +203,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 size=3
 stride=2
 pad=1
@@ -186,6 +211,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 size=3
 stride=1
 pad=1
@@ -193,18 +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]
@@ -212,7 +245,7 @@
 coords=4
 rescore=1
 side=7
-num=2
+num=3
 softmax=0
 sqrt=1
 jitter=.2

--
Gitblit v1.10.0