From 1c39aeca008ced62dae8245f8d6f9963f4c97b47 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 12 Apr 2018 11:12:38 +0000 Subject: [PATCH] Try to fix --- src/parser.c | 6 ++++-- 1 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/parser.c b/src/parser.c index 5a111bf..651671b 100644 --- a/src/parser.c +++ b/src/parser.c @@ -268,11 +268,13 @@ char *a = option_find_str(options, "mask", 0); int *mask = parse_yolo_mask(a, &num); - layer l = make_yolo_layer(params.batch, params.w, params.h, num, total, mask, classes); + int max_boxes = option_find_int_quiet(options, "max", 30); + layer l = make_yolo_layer(params.batch, params.w, params.h, num, total, mask, classes, max_boxes); assert(l.outputs == params.inputs); - l.max_boxes = option_find_int_quiet(options, "max", 90); + //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