From 2a9f7e44ce1b73d3d56ef83f83e94f074ecac3f9 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Fri, 06 Apr 2018 17:34:38 +0000 Subject: [PATCH] Added automatic AVX support - speedup +20% on CPU x86_64 Intel Skylake --- 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