Fix handling grey input frames

WillB97 · May 21, 2024 · 6464c09 · 6464c09
1 parent d6c8113
commit 6464c09
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/april_vision/utils.py b/april_vision/utils.py
@@ -29,6 +29,8 @@ def from_colour_frame(
         """Load frame from a colour image in a numpy array."""
         if colourspace is not None:
             grey_frame = cv2.cvtColor(colour_frame, colourspace)
+        else:
+            grey_frame = colour_frame.copy()
 
         return cls(
             grey_frame=grey_frame,