From 8b9af80a5b3bdb6ec17c4fa2cfa14ce4676ef1c3 Mon Sep 17 00:00:00 2001
From: Alexey <AlexeyAB@users.noreply.github.com>
Date: Fri, 18 May 2018 20:27:59 +0000
Subject: [PATCH] Merge pull request #851 from NadeenUdantha/patch-1
---
src/box.h | 15 ++++++++++++++-
1 files changed, 14 insertions(+), 1 deletions(-)
diff --git a/src/box.h b/src/box.h
index f9ca07e..9553b12 100644
--- a/src/box.h
+++ b/src/box.h
@@ -1,6 +1,11 @@
#ifndef BOX_H
#define BOX_H
+#ifdef _DEBUG
+#include <stdlib.h>
+#include <crtdbg.h>
+#endif
+
#ifdef YOLODLL_EXPORTS
#if defined(_MSC_VER)
#define YOLODLL_API __declspec(dllexport)
@@ -30,10 +35,14 @@
float *mask;
float objectness;
int sort_class;
+} detection;
+
+typedef struct detection_with_class {
+ detection det;
// The most probable class id: the best class index in this->prob.
// Is filled temporary when processing results, otherwise not initialized
int best_class;
-} detection;
+} detection_with_class;
box float_to_box(float *f);
float box_iou(box a, box b);
@@ -46,4 +55,8 @@
box decode_box(box b, box anchor);
box encode_box(box b, box anchor);
+// Creates array of detections with prob > thresh and fills best_class for them
+// Return number of selected detections in *selected_detections_num
+detection_with_class* get_actual_detections(detection *dets, int dets_num, float thresh, int* selected_detections_num);
+
#endif
--
Gitblit v1.10.0