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/resnet50_yolo.cfg | 32 ++++++-------------------------- 1 files changed, 6 insertions(+), 26 deletions(-) diff --git a/build/darknet/x64/resnet50_yolo.cfg b/build/darknet/x64/resnet50_yolo.cfg index 4a039dd..9a91193 100644 --- a/build/darknet/x64/resnet50_yolo.cfg +++ b/build/darknet/x64/resnet50_yolo.cfg @@ -1,18 +1,15 @@ [net] -# Training batch=64 -subdivisions=16 - -# Testing -#batch=1 -#subdivisions=1 - +subdivisions=8 height=416 width=416 -#max_crop=448 channels=3 momentum=0.9 decay=0.0005 +angle=0 +saturation = 1.5 +exposure = 1.5 +hue=.1 learning_rate=0.0001 max_batches = 45000 @@ -20,12 +17,6 @@ steps=100,25000,35000 scales=10,.1,.1 -angle=7 -hue=.1 -saturation=.75 -exposure=.75 -aspect=.75 - [convolutional] batch_normalize=1 filters=64 @@ -488,7 +479,7 @@ [shortcut] from=-4 activation=leaky - +#stopbackward=1 @@ -502,17 +493,6 @@ [convolutional] -batch_normalize=1 -size=3 -stride=1 -pad=1 -filters=1024 -activation=leaky - - - - -[convolutional] size=1 stride=1 pad=1 -- Gitblit v1.10.0