Small bug fixes

point_tracker/tracker2d.py

@@ -58,14 +58,18 @@ class Tracker2D:
     """
     Class to track 2d coordinates of blobs in images
     """
-    def __init__(self, image):
+    def __init__(self, images, frames):
         """
 
+        Args:
+            images: List of images (numpy arrays)
+            frames: List of frame numbers (int)
         """
 
         self.show_plot = False
 
-        self.images, self.frames = [], []
+        self.images = images  # type: List[np.array()]
+        self.frames = frames  # type: List[int]
 
         self.points = defaultdict(list)