From a7af0709cc4e032d4e0fee5a3fc4d6a6865d62a6 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Tue, 13 Dec 2016 12:55:55 +0000
Subject: [PATCH] Added: train_voc.cmd

---
 cfg/yolo.cfg |  370 +++++++++++++++++++++++++++++-----------------------
 1 files changed, 205 insertions(+), 165 deletions(-)

diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg
index eef0b69..fda339a 100644
--- a/cfg/yolo.cfg
+++ b/cfg/yolo.cfg
@@ -1,204 +1,244 @@
 [net]
-batch=64
-subdivisions=64
-height=448
-width=448
+batch=1
+subdivisions=1
+width=416
+height=416
 channels=3
-learning_rate=0.01
 momentum=0.9
 decay=0.0005
-
-[crop]
-crop_width=448
-crop_height=448
-flip=0
 angle=0
-saturation = 2
-exposure = 2
+saturation = 1.5
+exposure = 1.5
+hue=.1
+
+learning_rate=0.001
+max_batches = 120000
+policy=steps
+steps=-1,100,80000,100000
+scales=.1,10,.1,.1
 
 [convolutional]
+batch_normalize=1
+filters=32
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
 filters=64
-size=7
-stride=2
-pad=1
-activation=ramp
-
-[convolutional]
-filters=192
 size=3
-stride=2
+stride=1
 pad=1
-activation=ramp
+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
 pad=1
-activation=ramp
+activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=3
+stride=1
+pad=1
+activation=leaky
+
+[maxpool]
+size=2
 stride=2
-pad=1
-activation=ramp
 
 [convolutional]
-filters=128
-size=1
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=256
-size=3
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=128
-size=1
-stride=1
-pad=1
-activation=ramp
-
-[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
+
+[maxpool]
+size=2
 stride=2
-pad=1
-activation=ramp
 
 [convolutional]
-filters=256
-size=1
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=512
-size=3
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=256
-size=1
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=512
-size=3
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=256
-size=1
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=512
-size=3
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=256
-size=1
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=512
-size=3
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-filters=256
-size=1
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
+batch_normalize=1
 filters=1024
 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
+
+[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
+
+
+#######
+
+[convolutional]
+batch_normalize=1
+size=3
+stride=1
+pad=1
+filters=1024
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+size=3
+stride=1
+pad=1
+filters=1024
+activation=leaky
+
+[route]
+layers=-9
+
+[reorg]
 stride=2
-pad=1
-activation=ramp
+
+[route]
+layers=-1,-3
 
 [convolutional]
-filters=512
+batch_normalize=1
+size=3
+stride=1
+pad=1
+filters=1024
+activation=leaky
+
+[convolutional]
 size=1
 stride=1
 pad=1
-activation=ramp
+filters=425
+activation=linear
 
-[convolutional]
-filters=1024
-size=3
-stride=1
-pad=1
-activation=ramp
-
-[convolutional]
-size=3
-stride=1
-pad=1
-filters=1024
-activation=ramp
-
-[convolutional]
-size=3
-stride=2
-pad=1
-filters=1024
-activation=ramp
-
-[convolutional]
-size=3
-stride=1
-pad=1
-filters=1024
-activation=ramp
-
-[convolutional]
-size=3
-stride=1
-pad=1
-filters=1024
-activation=ramp
-
-[connected]
-output=4096
-activation=ramp
-
-[dropout]
-probability=.5
-
-[connected]
-output=1225
-activation=logistic
-
-[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=0
-joint=0
-objectness=1
+num=5
+softmax=1
+jitter=.2
+rescore=1
 
+object_scale=5
+noobject_scale=1
+class_scale=1
+coord_scale=1
+
+absolute=1
+thresh = .6
+random=0

--
Gitblit v1.10.0