From 0222895c0697b5158b05d3bbe01f0a88a3ccd576 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Sun, 27 Nov 2016 04:08:16 +0000 Subject: [PATCH] :crossed_swords: IS THIS NOT WHY YOU ARE HERE :crossed_swords: --- cfg/yolo.cfg | 262 ++++++++++++++++++++++++++++------------------------ 1 files changed, 141 insertions(+), 121 deletions(-) diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg index 1f69856..fda339a 100644 --- a/cfg/yolo.cfg +++ b/cfg/yolo.cfg @@ -1,31 +1,25 @@ [net] batch=1 subdivisions=1 -height=448 -width=448 +width=416 +height=416 channels=3 momentum=0.9 decay=0.0005 +angle=0 +saturation = 1.5 +exposure = 1.5 +hue=.1 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] -filters=64 -size=7 -stride=2 -pad=1 -activation=leaky - -[maxpool] -size=2 -stride=2 - -[convolutional] -filters=192 +batch_normalize=1 +filters=32 size=3 stride=1 pad=1 @@ -36,6 +30,55 @@ stride=2 [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 @@ -43,13 +86,27 @@ activation=leaky [convolutional] +batch_normalize=1 filters=256 size=3 stride=1 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 @@ -57,6 +114,23 @@ 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 @@ -68,87 +142,7 @@ stride=2 [convolutional] -filters=256 -size=1 -stride=1 -pad=1 -activation=leaky - -[convolutional] -filters=512 -size=3 -stride=1 -pad=1 -activation=leaky - -[convolutional] -filters=256 -size=1 -stride=1 -pad=1 -activation=leaky - -[convolutional] -filters=512 -size=3 -stride=1 -pad=1 -activation=leaky - -[convolutional] -filters=256 -size=1 -stride=1 -pad=1 -activation=leaky - -[convolutional] -filters=512 -size=3 -stride=1 -pad=1 -activation=leaky - -[convolutional] -filters=256 -size=1 -stride=1 -pad=1 -activation=leaky - -[convolutional] -filters=512 -size=3 -stride=1 -pad=1 -activation=leaky - -[convolutional] -filters=512 -size=1 -stride=1 -pad=1 -activation=leaky - -[convolutional] -filters=1024 -size=3 -stride=1 -pad=1 -activation=leaky - -[maxpool] -size=2 -stride=2 - -[convolutional] -filters=512 -size=1 -stride=1 -pad=1 -activation=leaky - -[convolutional] +batch_normalize=1 filters=1024 size=3 stride=1 @@ -156,6 +150,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=512 size=1 stride=1 @@ -163,15 +158,34 @@ 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 @@ -179,13 +193,24 @@ activation=leaky [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 size=3 stride=1 pad=1 @@ -193,32 +218,27 @@ activation=leaky [convolutional] -size=3 +size=1 stride=1 pad=1 -filters=1024 -activation=leaky - -[connected] -output=4096 -activation=leaky - -[connected] -output= 1470 +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=2 -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