github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 13:02:15 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

add common video extensions and get_video_paths similar to get_image_paths

Browse source
This commit is contained in:
mightbesimon 2023-06-25 05:23:39 +12:00
parent cc5fe74d6b
commit 7ee85c93de
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
core/pathex.py
View file

@ -2,6 +2,7 @@ from pathlib import Path
from os import scandir
image_extensions = [".jpg", ".jpeg", ".png", ".tif", ".tiff"]
video_extensions = [".mp4", ".mov", ".mpg", ".wmv", ".mkv", ".avi", ".avchd", ".flv", ".f4v", ".swf", "webm"]
def write_bytes_safe(p, bytes_data):
"""
@ -21,7 +22,7 @@ def scantree(path):
else:
yield entry
def get_image_paths(dir_path, image_extensions=image_extensions, subdirs=False, return_Path_class=False):
def get_extension_paths(dir_path, extensions, subdirs=False, return_Path_class=False):
dir_path = Path (dir_path)
result = []
@ -33,10 +34,16 @@ def get_image_paths(dir_path, image_extensions=image_extensions, subdirs=False,
gen = scandir(str(dir_path))
for x in gen:
if any(x.name.lower().endswith(ext) for ext in image_extensions): # listcomp is not needed, any() can handle gencomp
if any(x.name.lower().endswith(ext) for ext in extensions): # listcomp is not needed, any() can handle gencomp
result.append( Path(x) if return_Path_class else x.path ) # avoid unnecessary negative condition
return result # scandir should already be sorted
def get_image_paths(dir_path, image_extensions=image_extensions, subdirs=False, return_Path_class=False):
return get_extension_paths(dir_path, image_extensions, subdirs, return_Path_class)
def get_video_paths(dir_path, video_extensions=video_extensions, subdirs=False, return_Path_class=False):
return get_extension_paths(dir_path, video_extensions, subdirs, return_Path_class)
def get_image_unique_filestem_paths(dir_path, verbose_print_func=None):
result = get_image_paths(dir_path)
result_dup = set()