From c53e03348c65462bcba33f6352087dd6b9268e8f Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Wed, 16 Sep 2015 21:12:10 +0000 Subject: [PATCH] yolo working w/ regions --- cfg/darknet.cfg | 24 +++++++++++++----------- 1 files changed, 13 insertions(+), 11 deletions(-) diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg index f52ff3f..64aab1e 100644 --- a/cfg/darknet.cfg +++ b/cfg/darknet.cfg @@ -1,15 +1,17 @@ [net] -batch=128 +batch=256 subdivisions=1 height=256 width=256 channels=3 momentum=0.9 decay=0.0005 + learning_rate=0.01 -policy=poly -power=.5 -max_batches=600000 +policy=step +scale=.1 +step=100000 +max_batches=400000 [crop] crop_height=224 @@ -27,7 +29,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -38,7 +40,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -49,7 +51,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -60,7 +62,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -71,7 +73,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -82,7 +84,7 @@ activation=leaky [maxpool] -size=3 +size=2 stride=2 [convolutional] @@ -99,7 +101,7 @@ [connected] output=1000 -activation=linear +activation=leaky [softmax] -- Gitblit v1.10.0