Tracking with types: don't assign person to the bicycle!

Author: Nuzhny007

Tracker/Ctracker.cpp

@@ -92,15 +92,16 @@ void CTracker::Update(
         // -----------------------------------
         // Треки уже есть, составим матрицу расстояний
         // -----------------------------------
-		track_t maxCost = 0;
+        const track_t maxPossibleCost = grayFrame.cols * grayFrame.rows;
+        track_t maxCost = 0;
 		switch (m_distType)
         {
         case tracking::DistCenters:
             for (size_t i = 0; i < tracks.size(); i++)
             {
                 for (size_t j = 0; j < regions.size(); j++)
                 {
-                    auto dist = tracks[i]->CalcDist((regions[j].m_rect.tl() + regions[j].m_rect.br()) / 2);
+                    auto dist = tracks[i]->CheckType(regions[j].m_type) ? tracks[i]->CalcDist((regions[j].m_rect.tl() + regions[j].m_rect.br()) / 2) : maxPossibleCost;
 					Cost[i + j * N] = dist;
 					if (dist > maxCost)
 					{
@@ -115,7 +116,7 @@ void CTracker::Update(
             {
                 for (size_t j = 0; j < regions.size(); j++)
                 {
-					auto dist = tracks[i]->CalcDist(regions[j].m_rect);
+                    auto dist = tracks[i]->CheckType(regions[j].m_type) ? tracks[i]->CalcDist(regions[j].m_rect) : maxPossibleCost;
 					Cost[i + j * N] = dist;
 					if (dist > maxCost)
 					{
@@ -130,7 +131,7 @@ void CTracker::Update(
             {
                 for (size_t j = 0; j < regions.size(); j++)
                 {
-                    auto dist = tracks[i]->CalcDistJaccard(regions[j].m_rect);
+                    auto dist = tracks[i]->CheckType(regions[j].m_type) ? tracks[i]->CalcDistJaccard(regions[j].m_rect) : 1;
                     Cost[i + j * N] = dist;
                     if (dist > maxCost)
                     {

Tracker/track.cpp

@@ -85,6 +85,16 @@ track_t CTrack::CalcDistJaccard(const cv::Rect& r) const
     return 1 - intArea / unionArea;
 }
 
+///
+/// \brief CTrack::CheckType
+/// \param type
+/// \return
+///
+bool CTrack::CheckType(const std::string& type) const
+{
+    return m_lastRegion.m_type.empty() || type.empty() || (m_lastRegion.m_type == type);
+}
+
 ///
 /// \brief Update
 /// \param pt

Tracker/track.h

@@ -177,6 +177,7 @@ class CTrack
     track_t CalcDist(const Point_t& pt) const;
     track_t CalcDist(const cv::Rect& r) const;
     track_t CalcDistJaccard(const cv::Rect& r) const;
+    bool CheckType(const std::string& type) const;
 
     void Update(const CRegion& region, bool dataCorrect, size_t max_trace_length, cv::UMat prevFrame, cv::UMat currFrame);