Integration of the visual debug, with zero changes to the tracker's API.

[hercules2020/kcf.git] / src / kcf.cpp

diff --git a/src/kcf.cpp b/src/kcf.cpp
index 75119bd74ffcb5b5f0c3f9e00d348157e5ab1770..32cd9501408b71f3f7e90e3c14f661a7076c84a5 100644 (file)
--- a/src/kcf.cpp
+++ b/src/kcf.cpp
@@ -321,6 +321,39 @@ double KCF_Tracker::findMaxReponse(uint &max_idx, cv::Point2d &new_location) con
         }
     }
 #endif
+
+    if (m_visual_debug) {
+        cv::Mat all_responses(cv::Size(p_num_angles * 100, p_num_scales * 100),
+                              d.threadctxs[max_idx].response.plane(0).type(), cv::Scalar::all(0));
+        for (size_t i = 0; i < p_num_scales; ++i) {
+            for (size_t j = 0; j < p_num_angles; ++j) {
+                cv::Mat in_roi(all_responses, cv::Rect(j * 100, i * 100, 100, 100));
+                cv::Mat copy_response = d.threadctxs[p_num_angles * i + j].response.plane(0).clone();
+
+                copy_response = copy_response(cv::Rect(0, 0, copy_response.cols & -2, copy_response.rows & -2));
+
+                int cx = copy_response.cols / 2;
+                int cy = copy_response.rows / 2;
+                cv::Mat q0(copy_response, cv::Rect(0, 0, cx, cy));
+                cv::Mat q1(copy_response, cv::Rect(cx, 0, cx, cy));
+                cv::Mat q2(copy_response, cv::Rect(0, cy, cx, cy));
+                cv::Mat q3(copy_response, cv::Rect(cx, cy, cx, cy));
+                cv::Mat tmp;
+                q0.copyTo(tmp);
+                q3.copyTo(q0);
+                tmp.copyTo(q3);
+                q1.copyTo(tmp);
+                q2.copyTo(q1);
+                tmp.copyTo(q2);
+
+                copy_response.copyTo(in_roi);
+            }
+        }
+        cv::namedWindow("All responses", CV_WINDOW_AUTOSIZE);
+        cv::imshow("All responses", all_responses);
+        cv::waitKey();
+    }
+
     cv::Point2i &max_response_pt = IF_BIG_BATCH(d.threadctxs[0].max[max_idx].loc,        d.threadctxs[max_idx].max.loc);
     cv::Mat max_response_map     = IF_BIG_BATCH(d.threadctxs[0].response.plane(max_idx), d.threadctxs[max_idx].response.plane(0));