Adding logging in the Tracker class. !NOT TESTED!

Author: GBJim

iou_tracker.py

@@ -9,27 +9,32 @@
 from util import load_mot, iou
 
 class Tracker():
-    def __init__(self, sigma_l=0, sigma_h=0.5, sigma_iou=0.5, t_max=5):
+    def __init__(self, sigma_l=0, sigma_h=0.5, sigma_iou=0.5, t_max=5, verbose=False):
         self.sigma_l = sigma_l
         self.sigma_h = sigma_h
         self.sigma_iou = sigma_iou
         self.t_max = t_max
         self.frame = 0
         self.id_count = 0
         self.tracks_active = {}
+        self.verbose = verbose
 
     #Clear the old tracks
     def clean_old_tracks(self):
         target_frame = self.frame - self.t_max
         if self.tracks_active.has_key(target_frame):
+            if self.verbose:
+                print("[LOG]: Tracks Deleted: {}".format(self.tracks_active[target_frame].keys()))
             del(self.tracks_active[target_frame])
 
     #Retrieve tracks in an correct matching order
     def retrieve_tracks(self):
         tracks = []
         frames = range(self.frame, self.frame - self.t_max, -1)
-        for frame in frames:
+        for frame in frames: 
             if frame in self.tracks_active:
+                if self.verbose:
+                    print("[LOG]: Frame {} Tracks Retrieved: {}."format(frame,self.tracks_active[frame].keys()))
                 tracks += self.tracks_active[frame].items()
         return tracks
 
@@ -46,16 +51,23 @@ def track(self, detections):
                 # get det with highest iou
                 best_match = max(dets, key=lambda x: iou(track['bbox'], x['bbox']))
                 if iou(track['bbox'], best_match['bbox']) >= self.sigma_iou:
-                    self.tracks_active[self.frame][id_] = best_match                 
+                    self.tracks_active[self.frame][id_] = best_match
+                    if self.verbose:
+                        print("[LOG]: Tracke {} updated with bbox: {}".format(id_ , best_match['bbox']))
                     # remove from best matching detection from detections
                     del dets[dets.index(best_match)]
 
         #Create new tracks
         for det in dets:
             self.id_count += 1
             self.tracks_active[self.frame][self.id_count] = det
+            if self.verbose:
+                print("[LOG]: Tracke {} added with bbox: {}".format(self.id_count, best_match['bbox']))
+
 
         #Return the current tracks 
+        if self.verbose:
+            print("[LOG]: Trackes {} returned".format(self.tracks_active[self.frame].keys()))
         return self.tracks_active[self.frame]