Fix bug that can happen with fast processing dataset

2025-06-25 14:01:08 -06:00 · 2025-06-25 14:01:08 -06:00 · 24cd94929e
parent 19ea8ecc38
commit 24cd94929e
1 changed files with 2 additions and 0 deletions
--- a/toolkit/data_transfer_object/data_loader.py
+++ b/toolkit/data_transfer_object/data_loader.py
@ -91,6 +91,8 @@ class FileItemDTO(
                except image_utils.UnknownImageFormat:
                    print_once(f'Warning: Some images in the dataset cannot be fast read. ' + \
                            f'This process is faster for png, jpeg')
+                    img = exif_transpose(Image.open(self.path))
+                    w, h = img.size
            else:
                img = exif_transpose(Image.open(self.path))
                w, h = img.size