From 103fa8d22238a5829273adcb5d66568b2d9742ca Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Mon, 18 May 2020 17:00:29 +0300 Subject: [PATCH] [yolo] max=200 by default for crowds --- src/parser.c | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/parser.c b/src/parser.c index a64a7060..edfbe4dc 100644 --- a/src/parser.c +++ b/src/parser.c @@ -401,7 +401,7 @@ layer parse_yolo(list *options, size_params params) int num = total; char *a = option_find_str(options, "mask", 0); int *mask = parse_yolo_mask(a, &num); - int max_boxes = option_find_int_quiet(options, "max", 90); + int max_boxes = option_find_int_quiet(options, "max", 200); layer l = make_yolo_layer(params.batch, params.w, params.h, num, total, mask, classes, max_boxes); if (l.outputs != params.inputs) { printf("Error: l.outputs == params.inputs \n"); @@ -506,7 +506,7 @@ int *parse_gaussian_yolo_mask(char *a, int *num) // Gaussian_YOLOv3 layer parse_gaussian_yolo(list *options, size_params params) // Gaussian_YOLOv3 { int classes = option_find_int(options, "classes", 20); - int max_boxes = option_find_int_quiet(options, "max", 90); + int max_boxes = option_find_int_quiet(options, "max", 200); int total = option_find_int(options, "num", 1); int num = total; @@ -600,7 +600,7 @@ layer parse_region(list *options, size_params params) int coords = option_find_int(options, "coords", 4); int classes = option_find_int(options, "classes", 20); int num = option_find_int(options, "num", 1); - int max_boxes = option_find_int_quiet(options, "max", 90); + int max_boxes = option_find_int_quiet(options, "max", 200); layer l = make_region_layer(params.batch, params.w, params.h, num, classes, coords, max_boxes); if (l.outputs != params.inputs) { @@ -665,7 +665,7 @@ detection_layer parse_detection(list *options, size_params params) layer.softmax = option_find_int(options, "softmax", 0); layer.sqrt = option_find_int(options, "sqrt", 0); - layer.max_boxes = option_find_int_quiet(options, "max",30); + layer.max_boxes = option_find_int_quiet(options, "max",200); layer.coord_scale = option_find_float(options, "coord_scale", 1); layer.forced = option_find_int(options, "forced", 0); layer.object_scale = option_find_float(options, "object_scale", 1);