From 71a9929af6c3d3ffb9527bb921c5cc4a20971ff6 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Sat, 18 Mar 2017 01:47:21 +0300 Subject: [PATCH] Fixed x & y coords less than 0 --- src/yolo_v2_class.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/yolo_v2_class.cpp b/src/yolo_v2_class.cpp index c58ebdac..fb069858 100644 --- a/src/yolo_v2_class.cpp +++ b/src/yolo_v2_class.cpp @@ -18,6 +18,7 @@ extern "C" { #include #include +#include #define FRAMES 3 @@ -182,8 +183,8 @@ YOLODLL_API std::vector Detector::detect(image_t img, float thresh) if (prob > thresh) { bbox_t bbox; - bbox.x = (b.x - b.w / 2.)*im.w; - bbox.y = (b.y - b.h / 2.)*im.h; + bbox.x = std::max((double)0, (b.x - b.w / 2.)*im.w); + bbox.y = std::max((double)0, (b.y - b.h / 2.)*im.h); bbox.w = b.w*im.w; bbox.h = b.h*im.h; bbox.obj_id = obj_id;