Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make file name extension check case insensitive, so .JPG or .PNG files will also be detected

This commit is contained in:
Jan Gerritsen 2023-01-10 11:54:26 +01:00
parent 97a8c69451
commit fb73c16515
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
data/data_loader.py
View File

@ -231,7 +231,7 @@ class DataLoaderMultiAspect():
current = os.path.join(recurse_root, f) current = os.path.join(recurse_root, f)
if os.path.isfile(current): if os.path.isfile(current):
ext = os.path.splitext(f)[1] ext = os.path.splitext(f)[1].lower()
if ext in ['.jpg', '.jpeg', '.png', '.bmp', '.webp', '.jfif']: if ext in ['.jpg', '.jpeg', '.png', '.bmp', '.webp', '.jfif']:
# add image multiplyrepeats number of times # add image multiplyrepeats number of times
for _ in range(multiply): for _ in range(multiply):