ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make data loader resiliant to bad headers in meta

Browse Source
This commit is contained in:
Jaret Burkett
2023-08-29 18:56:06 -06:00
parent 836fee47a6
commit d401348c2e
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
toolkit/dataloader_mixins.py
View File

@@ -244,7 +244,12 @@ class ImageProcessingDTOMixin:
transform: Union[None, transforms.Compose] transform: Union[None, transforms.Compose]
): ):
# todo make sure this matches # 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 w, h = img.size
if w > h and self.scale_to_width < self.scale_to_height: if w > h and self.scale_to_width < self.scale_to_height:
# throw error, they should match # throw error, they should match