From 8bcdee86585f496afe1a8a38d608ea0504a11243 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 01 Sep 2015 18:22:03 +0000
Subject: [PATCH] Some bug fixes, random stuff
---
src/yolo.c | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/src/yolo.c b/src/yolo.c
index 13f0824..9bf96de 100644
--- a/src/yolo.c
+++ b/src/yolo.c
@@ -138,6 +138,7 @@
pthread_join(load_thread, 0);
free_data(buffer);
+ args.background = background;
load_thread = load_data_in_thread(args);
}
@@ -283,7 +284,7 @@
int w = val[t].w;
int h = val[t].h;
convert_yolo_detections(predictions, classes, objectness, background, num_boxes, w, h, thresh, probs, boxes);
- if (nms) do_nms(boxes, probs, num_boxes, classes, iou_thresh);
+ if (nms) do_nms(boxes, probs, num_boxes*num_boxes, classes, iou_thresh);
print_yolo_detections(fps, id, boxes, probs, num_boxes, classes, w, h);
free(id);
free_image(val[t]);
--
Gitblit v1.10.0