diff --git a/data/data_loader.py b/data/data_loader.py index 269f7af..a5591a1 100644 --- a/data/data_loader.py +++ b/data/data_loader.py @@ -231,7 +231,7 @@ class DataLoaderMultiAspect(): current = os.path.join(recurse_root, f) 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']: # add image multiplyrepeats number of times for _ in range(multiply):