in data/datasets.py [0:0]
def _process_images(self, images):
processed_images = []
for image in images:
if isinstance(image, Image.Image):
if image.mode != 'RGB':
image = image.convert('RGB')
processed_image = self.image_processor(image)
processed_images.append(processed_image)
else:
raise ValueError("Error processing image")
return processed_images