Merge pull request Smorodov#58 from Nuzhny007/master

Fix bug with tracking a targets biggest than half frame size
wwfzs1990 · Nov 21, 2017 · 41c61f0 · 41c61f0
2 parents 00b6d7f + 16d4725
commit 41c61f0
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/Tracker/track.cpp b/Tracker/track.cpp
@@ -220,7 +220,15 @@ void CTrack::RectUpdate(
 #ifdef USE_OCV_KCF
         if (!dataCorrect)
         {
-            cv::Size roiSize(currFrame.cols / 2, currFrame.rows / 2);
+            cv::Size roiSize(std::max(2 * m_predictionRect.width, currFrame.cols / 2), std::min(2 * m_predictionRect.height, currFrame.rows / 2));
+            if (roiSize.width > currFrame.cols)
+            {
+                roiSize.width = currFrame.cols;
+            }
+            if (roiSize.height > currFrame.rows)
+            {
+                roiSize.height = currFrame.rows;
+            }
             cv::Point roiTL(m_predictionRect.x + m_predictionRect.width / 2 - roiSize.width / 2, m_predictionRect.y + m_predictionRect.height / 2 - roiSize.height / 2);
             cv::Rect roiRect(roiTL, roiSize);
             Clamp(roiRect.x, roiRect.width, currFrame.cols);