From 71a9929af6c3d3ffb9527bb921c5cc4a20971ff6 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 17 Mar 2017 22:47:21 +0000
Subject: [PATCH] Fixed x & y coords less than 0
---
src/yolo_v2_class.cpp | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/yolo_v2_class.cpp b/src/yolo_v2_class.cpp
index c58ebda..fb06985 100644
--- a/src/yolo_v2_class.cpp
+++ b/src/yolo_v2_class.cpp
@@ -18,6 +18,7 @@
#include <vector>
#include <iostream>
+#include <algorithm>
#define FRAMES 3
@@ -182,8 +183,8 @@
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;
--
Gitblit v1.10.0