From 7b324bcc3b31d1de8756314d9c55ecdbdc710822 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Tue, 21 Mar 2023 21:10:34 +0900 Subject: [PATCH] support extensions of image files with uppercases --- library/train_util.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/library/train_util.py b/library/train_util.py index ec352a1e..97f5a702 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -73,8 +73,7 @@ DEFAULT_LAST_OUTPUT_NAME = "last" # region dataset -IMAGE_EXTENSIONS = [".png", ".jpg", ".jpeg", ".webp", ".bmp"] -# , ".PNG", ".JPG", ".JPEG", ".WEBP", ".BMP"] # Linux? +IMAGE_EXTENSIONS = [".png", ".jpg", ".jpeg", ".webp", ".bmp", ".PNG", ".JPG", ".JPEG", ".WEBP", ".BMP"] class ImageInfo: @@ -679,7 +678,7 @@ class BaseDataset(torch.utils.data.Dataset): # ちょっと速くした print("caching latents.") - image_infos= list(self.image_data.values()) + image_infos = list(self.image_data.values()) # sort by resolution image_infos.sort(key=lambda info: info.bucket_reso[0] * info.bucket_reso[1]) @@ -713,7 +712,7 @@ class BaseDataset(torch.utils.data.Dataset): if len(batch) > 0: batches.append(batch) - # iterate batches + # iterate batches for batch in tqdm(batches, smoothing=1, total=len(batches)): images = [] for info in batch: