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/alexnet.cfg | 60 +++++++++++++++++++++++++++++++++--------------------------- 1 files changed, 33 insertions(+), 27 deletions(-) diff --git a/cfg/alexnet.cfg b/cfg/alexnet.cfg index 79c5351..7e5a9b2 100644 --- a/cfg/alexnet.cfg +++ b/cfg/alexnet.cfg @@ -1,85 +1,91 @@ [net] batch=128 -subdivisions=128 -height=256 -width=256 +subdivisions=1 +height=227 +width=227 channels=3 -learning_rate=0.01 momentum=0.9 decay=0.0005 +max_crop=256 -[crop] -crop_height=224 -crop_width=224 -flip=1 -angle=0 -saturation=1 -exposure=1 +learning_rate=0.01 +policy=poly +power=4 +max_batches=800000 + +angle=7 +hue = .1 +saturation=.75 +exposure=.75 +aspect=.75 [convolutional] -filters=64 +filters=96 size=11 stride=4 pad=0 -activation=ramp +activation=relu [maxpool] size=3 stride=2 +padding=0 [convolutional] -filters=192 +filters=256 size=5 stride=1 pad=1 -activation=ramp +activation=relu [maxpool] size=3 stride=2 +padding=0 [convolutional] filters=384 size=3 stride=1 pad=1 -activation=ramp +activation=relu + +[convolutional] +filters=384 +size=3 +stride=1 +pad=1 +activation=relu [convolutional] filters=256 size=3 stride=1 pad=1 -activation=ramp - -[convolutional] -filters=256 -size=3 -stride=1 -pad=1 -activation=ramp +activation=relu [maxpool] size=3 stride=2 +padding=0 [connected] output=4096 -activation=ramp +activation=relu [dropout] probability=.5 [connected] output=4096 -activation=ramp +activation=relu [dropout] probability=.5 [connected] output=1000 -activation=ramp +activation=linear [softmax] groups=1 -- Gitblit v1.10.0