From 35cc0aaa15b991b348cc8d9623eed5d4f8a1e435 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 30 Mar 2018 12:51:54 +0000
Subject: [PATCH] Added max_boxes for yolo v3

---
 src/parser.c |    5 +++--
 1 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/parser.c b/src/parser.c
index 5a111bf..4de8aeb 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -268,10 +268,11 @@
 
 	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.ignore_thresh = option_find_float(options, "ignore_thresh", .5);

--
Gitblit v1.10.0