From 846b3b4366c8d1a1ab7215db3ce4f4180ea53cb5 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 30 Jul 2015 23:19:14 +0000
Subject: [PATCH] Added COCO, fixed memory leaks
---
src/box.c | 20 ++++++++++++++++++++
1 files changed, 20 insertions(+), 0 deletions(-)
diff --git a/src/box.c b/src/box.c
index cce56bd..0518c05 100644
--- a/src/box.c
+++ b/src/box.c
@@ -211,3 +211,23 @@
return dd;
}
+void do_nms(box *boxes, float **probs, int num_boxes, int classes, float thresh)
+{
+ int i, j, k;
+ for(i = 0; i < num_boxes*num_boxes; ++i){
+ int any = 0;
+ for(k = 0; k < classes; ++k) any = any || (probs[i][k] > 0);
+ if(!any) {
+ continue;
+ }
+ for(j = i+1; j < num_boxes*num_boxes; ++j){
+ if (box_iou(boxes[i], boxes[j]) > thresh){
+ for(k = 0; k < classes; ++k){
+ if (probs[i][k] < probs[j][k]) probs[i][k] = 0;
+ else probs[j][k] = 0;
+ }
+ }
+ }
+ }
+}
+
--
Gitblit v1.10.0