From d09b987b0f2e16ac3e84e81611e2f6b1cce63c84 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Thu, 16 Mar 2017 13:21:49 +0300 Subject: [PATCH] Fix in bbox_t coords from (float) to (unsigned int) --- src/yolo_v2_class.hpp | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/yolo_v2_class.hpp b/src/yolo_v2_class.hpp index a846f4aa..f1c6576c 100644 --- a/src/yolo_v2_class.hpp +++ b/src/yolo_v2_class.hpp @@ -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 detect(std::string image_filename, float thresh = 0.2); - YOLODLL_API std::vector detect(image_t img, float thresh = 0.2); - #ifdef OPENCV - std::vector detect(cv::Mat mat, float thresh = 0.2) { + std::vector detect(cv::Mat mat, float thresh = 0.2) + { std::shared_ptr image_ptr(new image_t, [](image_t *img) { free_image(*img); } ); *image_ptr = mat_to_image(mat); return detect(*image_ptr, thresh);