New filter for trajectory

hookttg · Jul 6, 2022 · bea1c3e · bea1c3e
1 parent ca760de
commit bea1c3e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/Tracker/trajectory.h b/src/Tracker/trajectory.h
@@ -213,9 +213,14 @@ struct TrackingObject
     /// \param sizeRatio
     /// \return
     ///
-	bool IsRobust(int minTraceSize, float minRawRatio, cv::Size2f sizeRatio) const
+	bool IsRobust(int minTraceSize, float minRawRatio, cv::Size2f sizeRatio, size_t lastDetectsCount = 0) const
 	{
 		m_lastRobust = m_trace.size() > static_cast<size_t>(minTraceSize);
+		if (lastDetectsCount)
+		{
+			size_t raws = m_trace.GetRawCount(lastDetectsCount);
+			m_lastRobust = (raws > 0);
+		}
 		m_lastRobust &= m_trace.GetRawCount(m_trace.size() - 1) / static_cast<float>(m_trace.size()) > minRawRatio;
 		if (sizeRatio.width + sizeRatio.height > 0)
 		{