From 943f6e874b819271a87665cf41199388380989a0 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 05 Apr 2018 20:27:02 +0000 Subject: [PATCH] Added Focal Loss to yolo-layer --- src/parser.c | 1 + 1 files changed, 1 insertions(+), 0 deletions(-) diff --git a/src/parser.c b/src/parser.c index 4de8aeb..651671b 100644 --- a/src/parser.c +++ b/src/parser.c @@ -274,6 +274,7 @@ //l.max_boxes = option_find_int_quiet(options, "max", 90); l.jitter = option_find_float(options, "jitter", .2); + l.focal_loss = option_find_int_quiet(options, "focal_loss", 0); l.ignore_thresh = option_find_float(options, "ignore_thresh", .5); l.truth_thresh = option_find_float(options, "truth_thresh", 1); -- Gitblit v1.10.0