From 8f1b4e0962857d402f9d017fcbf387ef0eceb7c4 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Thu, 01 Sep 2016 23:48:41 +0000 Subject: [PATCH] updates and things --- cfg/yolo.cfg | 216 +++++++++++++++++++++++++++++++++-------------------- 1 files changed, 134 insertions(+), 82 deletions(-) diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg index 40f87c4..c4f415c 100644 --- a/cfg/yolo.cfg +++ b/cfg/yolo.cfg @@ -1,205 +1,257 @@ [net] -batch=64 -subdivisions=64 +batch=1 +subdivisions=1 height=448 width=448 channels=3 -learning_rate=0.01 momentum=0.9 decay=0.0005 -seen = 0 +saturation=1.5 +exposure=1.5 +hue=.1 -[crop] -crop_width=448 -crop_height=448 -flip=0 -angle=0 -saturation = 2 -exposure = 2 +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 pad=1 -activation=ramp +activation=leaky + +[maxpool] +size=2 +stride=2 [convolutional] +batch_normalize=1 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=1 stride=1 pad=1 -activation=ramp +activation=leaky [convolutional] +batch_normalize=1 filters=256 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=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] -filters=512 -size=3 -stride=2 -pad=1 -activation=ramp - -[convolutional] +batch_normalize=1 filters=256 size=1 stride=1 pad=1 -activation=ramp +activation=leaky [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 pad=1 -activation=ramp +activation=leaky [convolutional] +batch_normalize=1 filters=256 size=1 stride=1 pad=1 -activation=ramp +activation=leaky [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 pad=1 -activation=ramp +activation=leaky [convolutional] +batch_normalize=1 filters=256 size=1 stride=1 pad=1 -activation=ramp +activation=leaky [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 pad=1 -activation=ramp +activation=leaky [convolutional] +batch_normalize=1 filters=256 size=1 stride=1 pad=1 -activation=ramp +activation=leaky [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 pad=1 -activation=ramp +activation=leaky [convolutional] -filters=256 -size=1 -stride=1 -pad=1 -activation=ramp - -[convolutional] -filters=1024 -size=3 -stride=2 -pad=1 -activation=ramp - -[convolutional] +batch_normalize=1 filters=512 size=1 stride=1 pad=1 -activation=ramp +activation=leaky [convolutional] +batch_normalize=1 filters=1024 size=3 stride=1 pad=1 -activation=ramp +activation=leaky + +[maxpool] +size=2 +stride=2 [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=ramp +activation=leaky [convolutional] +batch_normalize=1 size=3 stride=2 pad=1 filters=1024 -activation=ramp +activation=leaky [convolutional] +batch_normalize=1 size=3 stride=1 pad=1 filters=1024 -activation=ramp +activation=leaky [convolutional] +batch_normalize=1 size=3 stride=1 pad=1 filters=1024 -activation=ramp +activation=leaky -[connected] -output=4096 -activation=ramp +[local] +size=3 +stride=1 +pad=1 +filters=256 +activation=leaky [dropout] probability=.5 [connected] -output=1225 -activation=logistic +output= 1715 +activation=linear [detection] classes=20 coords=4 -rescore=0 -joint=1 -objectness = 0 -background=0 +rescore=1 +side=7 +num=3 +softmax=0 +sqrt=1 +jitter=.2 + +object_scale=1 +noobject_scale=.5 +class_scale=1 +coord_scale=5 + -- Gitblit v1.10.0