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

bug in file renamer

This commit is contained in:
Victor Hall 2022-11-11 00:37:12 -05:00
parent 1e136ff080
commit bc2001ac7f
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
environment.yaml
View File

@ -4,7 +4,6 @@ dependencies:
- aiofiles>=22.1.0 - aiofiles>=22.1.0
- colorama>=0.4.5 - colorama>=0.4.5
- aiohttp>=3.8.3 - aiohttp>=3.8.3
#- open_clip_torch>=1.26.12
- timm - timm
- fairscale==0.4.4 - fairscale==0.4.4
- transformers==4.19.2 - transformers==4.19.2

4
scripts/filename_replace.py
View File

@ -60,7 +60,7 @@ def get_replace_list(opt):
def rename_files(opt): def rename_files(opt):
find_list = get_replace_list(opt) find_list = get_replace_list(opt)
for idx, file in glob.iglob(opt.img_dir + "/*"): for idx, file in enumerate(glob.iglob(f"{opt.img_dir}/*")):
print(file) print(file)
if os.path.splitext(file)[1] in (".jpg", ".png", ".jpeg", ".gif", ".bmp", ".webp"): if os.path.splitext(file)[1] in (".jpg", ".png", ".jpeg", ".gif", ".bmp", ".webp"):
@ -74,7 +74,7 @@ def rename_files(opt):
try: try:
print(f"Renaming {file} to {new_filename}") print(f"Renaming {file} to {new_filename}")
if os.path.exists(new_filename): if os.path.exists(new_filename):
new_filename = new_filename + "_" + idx new_filename = f"{new_filename}_{idx}"
print(f"filename already exists, appended '_n' to {new_filename}") print(f"filename already exists, appended '_n' to {new_filename}")
try: try: