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

Fix bug that can happen with fast processing dataset

Browse Source
This commit is contained in:
Jaret Burkett
2025-06-25 14:01:08 -06:00
parent 19ea8ecc38
commit 24cd94929e
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/data_transfer_object/data_loader.py
View File

@@ -91,6 +91,8 @@ class FileItemDTO(
except image_utils.UnknownImageFormat: except image_utils.UnknownImageFormat:
print_once(f'Warning: Some images in the dataset cannot be fast read. ' + \ print_once(f'Warning: Some images in the dataset cannot be fast read. ' + \
f'This process is faster for png, jpeg') f'This process is faster for png, jpeg')
img = exif_transpose(Image.open(self.path))
w, h = img.size
else: else:
img = exif_transpose(Image.open(self.path)) img = exif_transpose(Image.open(self.path))
w, h = img.size w, h = img.size