From aebe937710ced03d03f73ab23f410f29685655c1 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Thu, 11 Aug 2016 18:54:24 +0000 Subject: [PATCH] what do you even write here? --- cfg/yolo.cfg | 44 ++++++++++++++++++++++++++++++++++++-------- 1 files changed, 36 insertions(+), 8 deletions(-) diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg index ee16726..6876eff 100644 --- a/cfg/yolo.cfg +++ b/cfg/yolo.cfg @@ -7,10 +7,10 @@ momentum=0.9 decay=0.0005 -learning_rate=0.001 +learning_rate=0.0005 policy=steps -steps=100,200,300,400,500,600,700,20000,30000 -scales=2,2,1.25,1.25,1.25,1.25,1.03,.1,.1 +steps=200,400,600,20000,30000 +scales=2.5,2,2,.1,.1 max_batches = 40000 [crop] @@ -20,8 +20,10 @@ angle=0 saturation = 1.5 exposure = 1.5 +noadjust=1 [convolutional] +batch_normalize=1 filters=64 size=7 stride=2 @@ -33,6 +35,7 @@ stride=2 [convolutional] +batch_normalize=1 filters=192 size=3 stride=1 @@ -44,6 +47,7 @@ stride=2 [convolutional] +batch_normalize=1 filters=128 size=1 stride=1 @@ -51,6 +55,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=256 size=3 stride=1 @@ -58,6 +63,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=256 size=1 stride=1 @@ -65,6 +71,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 @@ -76,6 +83,7 @@ stride=2 [convolutional] +batch_normalize=1 filters=256 size=1 stride=1 @@ -83,6 +91,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 @@ -90,6 +99,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=256 size=1 stride=1 @@ -97,6 +107,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 @@ -104,6 +115,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=256 size=1 stride=1 @@ -111,6 +123,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 @@ -118,6 +131,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=256 size=1 stride=1 @@ -125,6 +139,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 @@ -132,6 +147,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=512 size=1 stride=1 @@ -139,6 +155,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=1024 size=3 stride=1 @@ -150,6 +167,7 @@ stride=2 [convolutional] +batch_normalize=1 filters=512 size=1 stride=1 @@ -157,6 +175,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=1024 size=3 stride=1 @@ -164,6 +183,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=512 size=1 stride=1 @@ -171,6 +191,7 @@ activation=leaky [convolutional] +batch_normalize=1 filters=1024 size=3 stride=1 @@ -180,6 +201,7 @@ ####### [convolutional] +batch_normalize=1 size=3 stride=1 pad=1 @@ -187,6 +209,7 @@ activation=leaky [convolutional] +batch_normalize=1 size=3 stride=2 pad=1 @@ -194,6 +217,7 @@ activation=leaky [convolutional] +batch_normalize=1 size=3 stride=1 pad=1 @@ -201,29 +225,33 @@ activation=leaky [convolutional] +batch_normalize=1 size=3 stride=1 pad=1 filters=1024 activation=leaky -[connected] -output=4096 +[local] +size=3 +stride=1 +pad=1 +filters=256 activation=leaky [dropout] probability=.5 [connected] -output= 1470 +output= 1715 activation=linear -[region] +[detection] classes=20 coords=4 rescore=1 side=7 -num=2 +num=3 softmax=0 sqrt=1 jitter=.2 -- Gitblit v1.10.0