From 0b2be9d76a369e28fd1fc384b3952621921c9ae5 Mon Sep 17 00:00:00 2001 From: Joseph Redmon Date: Sat, 11 Jul 2015 22:52:43 -0700 Subject: [PATCH] Small version of YOLO, ~1GB RAM --- cfg/yolo-small.cfg | 205 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 205 insertions(+) create mode 100644 cfg/yolo-small.cfg diff --git a/cfg/yolo-small.cfg b/cfg/yolo-small.cfg new file mode 100644 index 00000000..e1be1ace --- /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