From c6afc7ff1499fbbe64069e1843d7929bd7ae2eaa Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 17 Nov 2016 20:18:24 +0000
Subject: [PATCH] :fire: :fire: yolo v2 :fire: :fire:

---
 cfg/yolo.cfg |  245 +++++++++++++++++++++++-------------------------
 1 files changed, 116 insertions(+), 129 deletions(-)

diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg
index c4f415c..4bf904c 100644
--- a/cfg/yolo.cfg
+++ b/cfg/yolo.cfg
@@ -1,36 +1,25 @@
 [net]
-batch=1
-subdivisions=1
-height=448
-width=448
+batch=64
+subdivisions=8
+height=416
+width=416
 channels=3
 momentum=0.9
 decay=0.0005
-saturation=1.5
-exposure=1.5
+angle=0
+saturation = 1.5
+exposure = 1.5
 hue=.1
 
-learning_rate=0.0005
+learning_rate=0.001
+max_batches = 120000
 policy=steps
-steps=200,400,600,20000,30000
-scales=2.5,2,2,.1,.1
-max_batches = 40000
+steps=-1,100,80000,100000
+scales=.1,10,.1,.1
 
 [convolutional]
 batch_normalize=1
-filters=64
-size=7
-stride=2
-pad=1
-activation=leaky
-
-[maxpool]
-size=2
-stride=2
-
-[convolutional]
-batch_normalize=1
-filters=192
+filters=32
 size=3
 stride=1
 pad=1
@@ -42,6 +31,54 @@
 
 [convolutional]
 batch_normalize=1
+filters=64
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=128
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=64
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=128
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=256
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
 filters=128
 size=1
 stride=1
@@ -56,6 +93,34 @@
 pad=1
 activation=leaky
 
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=512
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=256
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=512
+size=3
+stride=1
+pad=1
+activation=leaky
+
 [convolutional]
 batch_normalize=1
 filters=256
@@ -78,88 +143,12 @@
 
 [convolutional]
 batch_normalize=1
-filters=256
-size=1
-stride=1
-pad=1
-activation=leaky
-
-[convolutional]
-batch_normalize=1
-filters=512
-size=3
-stride=1
-pad=1
-activation=leaky
-
-[convolutional]
-batch_normalize=1
-filters=256
-size=1
-stride=1
-pad=1
-activation=leaky
-
-[convolutional]
-batch_normalize=1
-filters=512
-size=3
-stride=1
-pad=1
-activation=leaky
-
-[convolutional]
-batch_normalize=1
-filters=256
-size=1
-stride=1
-pad=1
-activation=leaky
-
-[convolutional]
-batch_normalize=1
-filters=512
-size=3
-stride=1
-pad=1
-activation=leaky
-
-[convolutional]
-batch_normalize=1
-filters=256
-size=1
-stride=1
-pad=1
-activation=leaky
-
-[convolutional]
-batch_normalize=1
-filters=512
-size=3
-stride=1
-pad=1
-activation=leaky
-
-[convolutional]
-batch_normalize=1
-filters=512
-size=1
-stride=1
-pad=1
-activation=leaky
-
-[convolutional]
-batch_normalize=1
 filters=1024
 size=3
 stride=1
 pad=1
 activation=leaky
 
-[maxpool]
-size=2
-stride=2
-
 [convolutional]
 batch_normalize=1
 filters=512
@@ -192,6 +181,7 @@
 pad=1
 activation=leaky
 
+
 #######
 
 [convolutional]
@@ -205,10 +195,19 @@
 [convolutional]
 batch_normalize=1
 size=3
+stride=1
+pad=1
+filters=1024
+activation=leaky
+
+[route]
+layers=-9
+
+[reorg]
 stride=2
-pad=1
-filters=1024
-activation=leaky
+
+[route]
+layers=-1,-3
 
 [convolutional]
 batch_normalize=1
@@ -219,39 +218,27 @@
 activation=leaky
 
 [convolutional]
-batch_normalize=1
-size=3
+size=1
 stride=1
 pad=1
-filters=1024
-activation=leaky
-
-[local]
-size=3
-stride=1
-pad=1
-filters=256
-activation=leaky
-
-[dropout]
-probability=.5
-
-[connected]
-output= 1715
+filters=425
 activation=linear
 
-[detection]
-classes=20
+[region]
+anchors = 0.738768,0.874946,  2.42204,2.65704,  4.30971,7.04493,  10.246,4.59428,  12.6868,11.8741
+bias_match=1
+classes=80
 coords=4
-rescore=1
-side=7
-num=3
-softmax=0
-sqrt=1
+num=5
+softmax=1
 jitter=.2
+rescore=1
 
-object_scale=1
-noobject_scale=.5
+object_scale=5
+noobject_scale=1
 class_scale=1
-coord_scale=5
+coord_scale=1
 
+absolute=1
+thresh = .6
+random=0

--
Gitblit v1.10.0