diff --git a/ldm/data/data_loader.py b/ldm/data/data_loader.py index debf7f0..f0abe20 100644 --- a/ldm/data/data_loader.py +++ b/ldm/data/data_loader.py @@ -1,8 +1,12 @@ import os from PIL import Image +import PIL import random from ldm.data.image_train_item import ImageTrainItem import ldm.data.aspects as aspects +from tqdm import tqdm + +PIL.Image.MAX_IMAGE_PIXELS = 933120000 class DataLoaderMultiAspect(): """ @@ -50,7 +54,7 @@ class DataLoaderMultiAspect(): """ decorated_image_train_items = [] - for pathname in image_paths: + for pathname in tqdm(image_paths): caption_from_filename = os.path.splitext(os.path.basename(pathname))[0].split("_")[0] txt_file_path = os.path.splitext(pathname)[0] + ".txt"