From 845ab7579685b6702c92c1088ec11e71bde51f3c Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Fri, 05 Aug 2016 22:27:07 +0000 Subject: [PATCH] some more stuff --- cfg/darknet.cfg | 51 ++++++++++++++++++++++++++++----------------------- 1 files changed, 28 insertions(+), 23 deletions(-) diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg index 2a99279..7c0d28a 100644 --- a/cfg/darknet.cfg +++ b/cfg/darknet.cfg @@ -1,22 +1,20 @@ [net] batch=128 subdivisions=1 -height=256 -width=256 +height=224 +width=224 channels=3 -learning_rate=0.01 momentum=0.9 decay=0.0005 +max_crop=320 -[crop] -crop_height=224 -crop_width=224 -flip=1 -angle=0 -saturation=1 -exposure=1 +learning_rate=0.1 +policy=poly +power=4 +max_batches=1600000 [convolutional] +batch_normalize=1 filters=16 size=3 stride=1 @@ -24,10 +22,11 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=32 size=3 stride=1 @@ -35,10 +34,11 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=64 size=3 stride=1 @@ -46,10 +46,11 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=128 size=3 stride=1 @@ -57,10 +58,11 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=256 size=3 stride=1 @@ -68,10 +70,11 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 @@ -79,26 +82,28 @@ activation=leaky [maxpool] -stride=2 size=2 +stride=2 [convolutional] +batch_normalize=1 filters=1024 size=3 stride=1 pad=1 activation=leaky -[avgpool] - -[dropout] -probability=.5 - -[connected] -output=1000 +[convolutional] +filters=1000 +size=1 +stride=1 +pad=1 activation=leaky +[avgpool] + [softmax] +groups=1 [cost] type=sse -- Gitblit v1.10.0