Fixed x & y coords less than 0

pull/72/head
AlexeyAB 8 years ago
parent 351996d9f0
commit 71a9929af6
  1. 5
      src/yolo_v2_class.cpp

@ -18,6 +18,7 @@ extern "C" {
#include <vector>
#include <iostream>
#include <algorithm>
#define FRAMES 3
@ -182,8 +183,8 @@ YOLODLL_API std::vector<bbox_t> 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;

Loading…
Cancel
Save