From a6c51e3b758aee7fd3a6f1d37daa8dcad4891e52 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 08 Mar 2018 22:42:58 +0000 Subject: [PATCH] Fixes for small objects. And fixes of densenet201_yolo.cfg, resnet50_yolo.cfg, resnet152_yolo.cfg --- build/darknet/x64/resnet152_yolo.cfg | 19 +++++++------------ 1 files changed, 7 insertions(+), 12 deletions(-) diff --git a/build/darknet/x64/resnet152_yolo.cfg b/build/darknet/x64/resnet152_yolo.cfg index 435695e..0ea6112 100644 --- a/build/darknet/x64/resnet152_yolo.cfg +++ b/build/darknet/x64/resnet152_yolo.cfg @@ -5,22 +5,17 @@ width=416 channels=3 momentum=0.9 -decay=0.0001 -#decay=0.0005 +decay=0.0005 angle=0 saturation = 1.5 exposure = 1.5 hue=.1 -learning_rate=0.01 -max_batches = 90000 +learning_rate=0.0001 +max_batches = 45000 policy=steps -steps=60000,80000 -scales=.1,.1 - -#learning_rate=0.0001 -#steps=100,25000,35000 -#scales=10,.1,.1 +steps=100,25000,35000 +scales=10,.1,.1 [convolutional] batch_normalize=1 @@ -1436,7 +1431,7 @@ [shortcut] from=-4 activation=leaky -stopbackward=1 +#stopbackward=1 @@ -1466,7 +1461,7 @@ softmax=1 jitter=.3 rescore=1 -focal_loss=1 +#focal_loss=1 object_scale=5 noobject_scale=1 -- Gitblit v1.10.0