diff --git a/ocv_corner_tracker.py b/ocv_corner_tracker.py index 14aa016..7da0736 100644 --- a/ocv_corner_tracker.py +++ b/ocv_corner_tracker.py @@ -100,8 +100,8 @@ class Corner: class CornerTracker: - def __init__(self, do_track, color=(0, 255, 0), name: str = 'CornerTracker'): - self.do_track = do_track + def __init__(self, _do_track, color=(0, 255, 0), name: str = 'CornerTracker'): + self.do_track = _do_track self.color = color self.name = name self.matching_tpl_bb = None @@ -330,7 +330,8 @@ if __name__ == '__main__': if args.scale is not None: scale = np.float32(args.scale) - do_track = args.track is not None + # Parse track + do_track = args.track # Let's go colors = [(255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 0, 255), (0, 255, 255), (255, 255, 255)]