raft working

src/orisys/sensor.py

@@ -166,8 +166,8 @@ class Sensor:
             return None
         self.frame = frame
         # Convert to grayscale
-        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
-
+        # gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+        gray = frame
         # Extract and stitch tiles
         self.stitcher.get_stitch_unit(gray)
         self._extract_tiles(gray)
@@ -215,7 +215,8 @@ class Sensor:
         if self.img is None:
             raise Exception("Warning: Get image first")
         
-        flow = self.__tracker.track(self.img)
+        # flow = self.__tracker.track(self.img)
+        flow = self.__tracker.track_raft(self.img)
         if hasattr(flow, 'download'):
             flow = flow.download()
         self.flow = flow