Merge pull request #51 from Kidel/main

fix file not found when `[` is in the filename
This commit is contained in:
Kohya S
2023-01-09 21:03:07 +09:00
committed by GitHub

View File

@@ -715,7 +715,7 @@ def debug_dataset(train_dataset):
def glob_images(dir, base):
img_paths = []
for ext in IMAGE_EXTENSIONS:
img_paths.extend(glob.glob(os.path.join(dir, base + ext)))
img_paths.extend(glob.glob(glob.escape(os.path.join(dir, base + ext))))
return img_paths
# endregion