Added grayscale reading in ModalAI

cmusatyalab · Feb 6, 2024 · 3ad7a5e · 3ad7a5e
1 parent 0eee67e
commit 3ad7a5e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/onboard/python/implementation/drones/ModalAISeekerDrone.py b/onboard/python/implementation/drones/ModalAISeekerDrone.py
@@ -242,7 +242,11 @@ def __init__(self, drone, ip):
     def run(self):
         try:
             while(self.isRunning):
-                ret, self.currentFrame = self.cap.read()
+                ret, frame = self.cap.read()
+                if len(frame.shape) < 3: # Grayscale
+                    self.currentFrame = cv2.cvtColor(frame, cv2.COLOR_GRAY2BGR)
+                else:
+                    self.currentFrame = frame
         except Exception as e:
             print(e)