From 5e448c00f32e80e13eead72318047ce8368aac9a Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 05 Mar 2018 16:59:29 +0000
Subject: [PATCH] Added Focal loss. Use focal_loss=1 in the [region] layer in your cfg-file. alpha = 0.5, gamma = 2

---
 cfg/vgg-16.cfg |    9 ++++-----
 1 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/cfg/vgg-16.cfg b/cfg/vgg-16.cfg
index e296eb9..2b6f702 100644
--- a/cfg/vgg-16.cfg
+++ b/cfg/vgg-16.cfg
@@ -1,21 +1,20 @@
 [net]
 batch=128
-subdivisions=8
+subdivisions=4
 height=256
 width=256
 channels=3
 learning_rate=0.00001
 momentum=0.9
-seen=0
 decay=0.0005
 
 [crop]
 crop_height=224
 crop_width=224
 flip=1
-exposure=2
-saturation=2
-angle=5
+exposure=1
+saturation=1
+angle=0
 
 [convolutional]
 filters=64

--
Gitblit v1.10.0