From baf3fcb491ee1a5e083fbdfdf2c28aaf36488e92 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 16 Feb 2018 23:26:08 +0000
Subject: [PATCH] Calculation TP, FP, FN, avg_IoU for default threshold during mAP calculation
---
src/yolo_v2_class.hpp | 3 +--
1 files changed, 1 insertions(+), 2 deletions(-)
diff --git a/src/yolo_v2_class.hpp b/src/yolo_v2_class.hpp
index 5aa3631..8abb9e2 100644
--- a/src/yolo_v2_class.hpp
+++ b/src/yolo_v2_class.hpp
@@ -193,8 +193,7 @@
void update_cur_bbox_vec(std::vector<bbox_t> _cur_bbox_vec)
{
cur_bbox_vec = _cur_bbox_vec;
- good_bbox_vec_flags.resize(cur_bbox_vec.size());
- for (auto &i : good_bbox_vec_flags) i = true;
+ good_bbox_vec_flags = std::vector<bool>(cur_bbox_vec.size(), true);
cv::Mat prev_pts, cur_pts_flow_cpu;
for (auto &i : cur_bbox_vec) {
--
Gitblit v1.10.0