mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 04:52:13 -07:00
625bcc212d
Linux does not guarantee filenames are returned in any specific order. This leads to exporting frames in random order, sorting them here makes the export run sequentially. Other portions of the program should remain unaffected, if not behave more consistently (E.G. get_first_file_by_stem). This mostly helpful during exporting. Say you are expecting to not have faces for frames 1000-2000, during your export all the "no faces for..." messages will appear in random order. Since you are expecting to see this you ignore them. If you are also (unexpectedly) missing a face for frame 3000 you will not head the warning since it's mixed up in all the warnings that you are expecting. With this patch export runs in sequential order, you'll see the messages all in a row for frames 1000-2000, then again at 3000. The user is much more likely to see and head the warning this way. This also allows you to force stop the export midway though and have a contiguous set of frames to encode and preview.
83 lines
2.4 KiB
Python
83 lines
2.4 KiB
Python
from pathlib import Path
|
|
from os import scandir
|
|
|
|
image_extensions = [".jpg", ".jpeg", ".png", ".tif", ".tiff"]
|
|
|
|
def get_image_paths(dir_path, image_extensions=image_extensions):
|
|
dir_path = Path (dir_path)
|
|
|
|
result = []
|
|
if dir_path.exists():
|
|
for x in list(scandir(str(dir_path))):
|
|
if any([x.name.lower().endswith(ext) for ext in image_extensions]):
|
|
result.append(x.path)
|
|
return sorted(result)
|
|
|
|
def get_image_unique_filestem_paths(dir_path, verbose_print_func=None):
|
|
result = get_image_paths(dir_path)
|
|
result_dup = set()
|
|
|
|
for f in result[:]:
|
|
f_stem = Path(f).stem
|
|
if f_stem in result_dup:
|
|
result.remove(f)
|
|
if verbose_print_func is not None:
|
|
verbose_print_func ("Duplicate filenames are not allowed, skipping: %s" % Path(f).name )
|
|
continue
|
|
result_dup.add(f_stem)
|
|
|
|
return sorted(result)
|
|
|
|
def get_file_paths(dir_path):
|
|
dir_path = Path (dir_path)
|
|
|
|
result = []
|
|
if dir_path.exists():
|
|
return [ x.path for x in list(scandir(str(dir_path))) if x.is_file() ]
|
|
return sorted(result)
|
|
|
|
def get_all_dir_names (dir_path):
|
|
dir_path = Path (dir_path)
|
|
|
|
result = []
|
|
if dir_path.exists():
|
|
return [ x.name for x in list(scandir(str(dir_path))) if x.is_dir() ]
|
|
|
|
return sorted(result)
|
|
|
|
def get_all_dir_names_startswith (dir_path, startswith):
|
|
dir_path = Path (dir_path)
|
|
startswith = startswith.lower()
|
|
|
|
result = []
|
|
if dir_path.exists():
|
|
for x in list(scandir(str(dir_path))):
|
|
if x.name.lower().startswith(startswith):
|
|
result.append ( x.name[len(startswith):] )
|
|
return sorted(result)
|
|
|
|
def get_first_file_by_stem (dir_path, stem, exts=None):
|
|
dir_path = Path (dir_path)
|
|
stem = stem.lower()
|
|
|
|
if dir_path.exists():
|
|
for x in sorted(list(scandir(str(dir_path)))):
|
|
if not x.is_file():
|
|
continue
|
|
xp = Path(x.path)
|
|
if xp.stem.lower() == stem and (exts is None or xp.suffix.lower() in exts):
|
|
return xp
|
|
|
|
return None
|
|
|
|
def move_all_files (src_dir_path, dst_dir_path):
|
|
paths = get_file_paths(src_dir_path)
|
|
for p in paths:
|
|
p = Path(p)
|
|
p.rename ( Path(dst_dir_path) / p.name )
|
|
|
|
def delete_all_files (dir_path):
|
|
paths = get_file_paths(dir_path)
|
|
for p in paths:
|
|
p = Path(p)
|
|
p.unlink()
|