Fixes for dataloader

2026-05-01 03:31:35 +00:00 · 2023-08-31 04:54:10 -06:00
parent addf024630
commit 4ca819a05e
2 changed files with 18 additions and 18 deletions
--- a/toolkit/dataloader_mixins.py
+++ b/toolkit/dataloader_mixins.py
@@ -251,8 +251,8 @@ class ImageProcessingDTOMixin:
            transform: Union[None, transforms.Compose]
    ):
        # todo make sure this matches
-        img = Image.open(self.path).convert('RGB')
        try:
+            img = Image.open(self.path).convert('RGB')
            img = exif_transpose(img)
        except Exception as e:
            print(f"Error: {e}")