Update tracker.py

lib/fish_eye/tracker.py

@@ -7,6 +7,7 @@ import numpy as np
 from fish_length import Fish_Length
 from sort import Sort
 from associative import Associate
+import lib
 
 class Tracker:
     def __init__(self, clip_info, algorithm=Sort, args={'max_age':1, 'min_hits':0, 'iou_threshold':0.05}, min_hits=3, reverse=False):
@@ -30,14 +31,10 @@ class Tracker:
             conf = 0
             min_score = 1000000
             print("algorithm", self.algorithm, type(self.algorithm))
-            print(type(self.algorithm) == Sort)
-            print(type(self.algorithm) == Associate)
-            print(isinstance(self.algorithm, Sort))
-            print(isinstance(self.algorithm, Associate))
-            print(self.algorithm is Sort)
-            print(self.algorithm is Associate)
-            print(type(self.algorithm) is Sort)
-            print(type(self.algorithm) is Associate)
+            print(type(self.algorithm) == lib.fish_eye.associative.Associate)
+            print(isinstance(self.algorithm, lib.fish_eye.associative.Associate))
+            print(self.algorithm is lib.fish_eye.associative.Associate)
+            print(type(self.algorithm) is lib.fish_eye.associative.Associate)
             if type(self.algorithm) == Sort:
                 print("SORT")
                 for det in dets: