Fix: convert images to RGB

luxonis · Oct 29, 2024 · 3ceab24 · 3ceab24
1 parent 1914f7d
commit 3ceab24
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/datadreamer/pipelines/generate_dataset_from_scratch.py b/datadreamer/pipelines/generate_dataset_from_scratch.py
@@ -552,7 +552,9 @@ def read_image_batch(image_batch, batch_num, batch_size):
                 batch_image_paths.append(image_path)
 
         else:
-            images = [Image.open(image_path) for image_path in image_batch]
+            images = [
+                Image.open(image_path).convert("RGB") for image_path in image_batch
+            ]
             batch_image_paths = image_batch
         return images, batch_image_paths