solving tracked objects

Author: tobias-gp-porsche

simple_filters/tracker.py

@@ -78,7 +78,7 @@ def update(self, states):
         ## Build the distance matrix and match objects
         # We build a matrix that contains the distances of the tracked objects 
         # with its predicted state (determined by the filter) and the new states which just came in
-        if number_of_tracked_objects > 0: 
+        if number_of_tracked_objects > 0 and number_of_states > 0: 
 
             # Calculate the distance matrix, the complexity is n^2
             distance_matrix = np.zeros((number_of_tracked_objects, number_of_states))

tests/test_single_object_tracking.py

@@ -30,4 +30,4 @@ def test_main(self):
             tracked_state = self.tracker.get_tracked_objects()
 
             self.assertEqual(1, len(tracked_state))
-            self.assertEqual(tracked_state[0].id, expected_tracking_id)
+            self.assertEqual(tracked_state[0].id, expected_tracking_id)