From 08d6910521d9aa05a11d0db154ae70874e47d0e1 Mon Sep 17 00:00:00 2001 From: Alexey <AlexeyAB@users.noreply.github.com> Date: Wed, 09 Aug 2017 21:06:24 +0000 Subject: [PATCH] Update Readme.md --- src/yolo_v2_class.hpp | 4 ++-- 1 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/yolo_v2_class.hpp b/src/yolo_v2_class.hpp index 4d4960c..bd3b4fd 100644 --- a/src/yolo_v2_class.hpp +++ b/src/yolo_v2_class.hpp @@ -54,7 +54,7 @@ YOLODLL_API int get_net_width(); YOLODLL_API int get_net_height(); - YOLODLL_API std::vector<bbox_t> tracking(std::vector<bbox_t> cur_bbox_vec, int const frames_story = 4); + YOLODLL_API std::vector<bbox_t> tracking(std::vector<bbox_t> cur_bbox_vec, int const frames_story = 6); #ifdef OPENCV std::vector<bbox_t> detect(cv::Mat mat, float thresh = 0.2, bool use_mean = false) @@ -64,7 +64,7 @@ cv::Mat det_mat; cv::resize(mat, det_mat, cv::Size(get_net_width(), get_net_height())); auto image_ptr = mat_to_image(det_mat); - auto detection_boxes = detect(*image_ptr, thresh); + auto detection_boxes = detect(*image_ptr, thresh, use_mean); float wk = (float)mat.cols / det_mat.cols, hk = (float)mat.rows / det_mat.rows; for (auto &i : detection_boxes) i.x*=wk, i.w*= wk, i.y*=hk, i.h*=hk; return detection_boxes; -- Gitblit v1.10.0