From 0b2be9d76a369e28fd1fc384b3952621921c9ae5 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Sun, 12 Jul 2015 05:52:43 +0000 Subject: [PATCH] Small version of YOLO, ~1GB RAM --- cfg/yolo-small.cfg | 205 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 205 insertions(+), 0 deletions(-) diff --git a/cfg/yolo-small.cfg b/cfg/yolo-small.cfg new file mode 100644 index 0000000..e1be1ac --- /dev/null +++ b/cfg/yolo-small.cfg @@ -0,0 +1,205 @@ +[net] +batch=64 +subdivisions=64 +height=448 +width=448 +channels=3 +learning_rate=0.01 +momentum=0.9 +decay=0.0005 +seen = 0 + +[crop] +crop_width=448 +crop_height=448 +flip=0 +angle=0 +saturation = 2 +exposure = 2 + +[convolutional] +filters=64 +size=7 +stride=2 +pad=1 +activation=ramp + +[convolutional] +filters=192 +size=3 +stride=2 +pad=1 +activation=ramp + +[convolutional] +filters=128 +size=1 +stride=1 +pad=1 +activation=ramp + +[convolutional] +filters=256 +size=3 +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] +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] +filters=1024 +size=3 +stride=2 +pad=1 +activation=ramp + +[convolutional] +filters=512 +size=1 +stride=1 +pad=1 +activation=ramp + +[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=512 +activation=ramp + +[connected] +output=2048 +activation=ramp + +[dropout] +probability=.5 + +[connected] +output=1225 +activation=logistic + +[detection] +classes=20 +coords=4 +rescore=1 +joint=1 +objectness = 0 +background=0 -- Gitblit v1.10.0