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

added VideoEd - operating video files by ffmpeg from python.

Browse source
This commit is contained in:
iperov 2019-03-02 14:58:22 +04:00
parent e0907423a9
commit 4196e962a8
6 changed files with 275 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

16
utils/Path_utils.py
View file

@ -3,7 +3,7 @@ from scandir import scandir
image_extensions = [".jpg", ".jpeg", ".png", ".tif", ".tiff"]
def get_image_paths(dir_path):
def get_image_paths(dir_path, image_extensions=image_extensions):
dir_path = Path (dir_path)
result = []
@ -38,3 +38,17 @@ def get_all_dir_names_startswith (dir_path, startswith):
if x.name.lower().startswith(startswith):
result.append ( x.name[len(startswith):] )
return 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 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