github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-14 00:53:48 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fixes

Browse source
This commit is contained in:
Colombo 2020-03-09 00:18:01 +04:00
parent eda6433936
commit d731930537
2 changed files with 6 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

4
core/pathex.py
View file

@ -21,7 +21,7 @@ def scantree(path):
else:
yield entry
def get_image_paths(dir_path, image_extensions=image_extensions, subdirs=False):
def get_image_paths(dir_path, image_extensions=image_extensions, subdirs=False, return_Path_class=False):
dir_path = Path (dir_path)
result = []
@ -34,7 +34,7 @@ def get_image_paths(dir_path, image_extensions=image_extensions, subdirs=False):
for x in list(gen):
if any([x.name.lower().endswith(ext) for ext in image_extensions]):
result.append(x.path)
result.append( x.path if not return_Path_class else Path(x.path) )
return sorted(result)
def get_image_unique_filestem_paths(dir_path, verbose_print_func=None):