Nadeen Udantha
2018-05-18 ae9fb2abbd8e2b7e079b56bf1cf39754d7c091ac
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) 
@@ -32,6 +37,13 @@
   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_with_class;
box float_to_box(float *f);
float box_iou(box a, box b);
float box_rmse(box a, box b);
@@ -43,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