From 3ceab248372b3f53711dc4f14cf001380c0504ea Mon Sep 17 00:00:00 2001 From: HonzaCuhel Date: Wed, 30 Oct 2024 00:56:55 +0100 Subject: [PATCH] Fix: convert images to RGB --- datadreamer/pipelines/generate_dataset_from_scratch.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/datadreamer/pipelines/generate_dataset_from_scratch.py b/datadreamer/pipelines/generate_dataset_from_scratch.py index 81908a5..9708f4f 100644 --- 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