From 5c1e8e3f48343d8944af1195e21f6f3b53ed848e Mon Sep 17 00:00:00 2001 From: Alexey <AlexeyAB@users.noreply.github.com> Date: Sat, 14 Apr 2018 23:08:50 +0000 Subject: [PATCH] Update Readme.md --- 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