From d401348c2e0a609950705d646c44a874139d7608 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 29 Aug 2023 18:56:06 -0600 Subject: [PATCH] Make data loader resiliant to bad headers in meta --- toolkit/dataloader_mixins.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/toolkit/dataloader_mixins.py b/toolkit/dataloader_mixins.py index 74567946..c25cfb14 100644 --- a/toolkit/dataloader_mixins.py +++ b/toolkit/dataloader_mixins.py @@ -244,7 +244,12 @@ class ImageProcessingDTOMixin: transform: Union[None, transforms.Compose] ): # todo make sure this matches - img = exif_transpose(Image.open(self.path)).convert('RGB') + img = Image.open(self.path).convert('RGB') + try: + img = exif_transpose(img) + except Exception as e: + print(f"Error: {e}") + print(f"Error loading image: {self.path}") w, h = img.size if w > h and self.scale_to_width < self.scale_to_height: # throw error, they should match