OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix in bbox_t coords from (float) to (unsigned int)

This commit is contained in:
AlexeyAB 2017-03-16 13:21:49 +03:00
parent 39912d2240
commit d09b987b0f
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/yolo_v2_class.hpp
View File

@ -15,7 +15,7 @@
#endif
struct bbox_t {
float x, y, w, h;
unsigned int x, y, w, h;
float prob;
unsigned int obj_id;
};
@ -36,12 +36,11 @@ public:
YOLODLL_API ~Detector();
YOLODLL_API std::vector<bbox_t> detect(std::string image_filename, float thresh = 0.2);
YOLODLL_API std::vector<bbox_t> detect(image_t img, float thresh = 0.2);
#ifdef OPENCV
std::vector<bbox_t> detect(cv::Mat mat, float thresh = 0.2) {
std::vector<bbox_t> detect(cv::Mat mat, float thresh = 0.2)
{
std::shared_ptr<image_t> image_ptr(new image_t, [](image_t *img) { free_image(*img); } );
*image_ptr = mat_to_image(mat);
return detect(*image_ptr, thresh);