AlexeyAB
2018-02-12 8df8b36f313688e3fb2b1240609c0cf12d8cf7b2
src/yolo_v2_class.hpp
@@ -144,8 +144,8 @@
};
#ifdef OPENCV
#ifdef TRACK_OPTFLOW
#if defined(TRACK_OPTFLOW) && defined(OPENCV)
#include <opencv2/cudaoptflow.hpp>
#include <opencv2/cudaimgproc.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) {
@@ -297,7 +296,8 @@
            float moved_y = cur_key_pt.y - prev_key_pt.y;
            if (abs(moved_x) < 100 && abs(moved_y) < 100 && good_bbox_vec_flags[i])
               if (err_cpu.at<float>(0, i) < flow_error && status_cpu.at<unsigned char>(0, i) != 0)
               if (err_cpu.at<float>(0, i) < flow_error && status_cpu.at<unsigned char>(0, i) != 0 &&
                  ((float)cur_bbox_vec[i].x + moved_x) > 0 && ((float)cur_bbox_vec[i].y + moved_y) > 0)
               {
                  cur_bbox_vec[i].x += moved_x + 0.5;
                  cur_bbox_vec[i].y += moved_y + 0.5;
@@ -324,9 +324,11 @@
class Tracker_optflow {};
#endif   // TRACK_OPTFLOW
#endif   // defined(TRACK_OPTFLOW) && defined(OPENCV)
#ifdef OPENCV
cv::Scalar obj_id_to_color(int obj_id) {
   int const colors[6][3] = { { 1,0,1 },{ 0,0,1 },{ 0,1,1 },{ 0,1,0 },{ 1,1,0 },{ 1,0,0 } };
   int const offset = obj_id * 123457 % 6;
@@ -469,4 +471,3 @@
   }
};
#endif   // OPENCV