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

minor logic optimisation

Browse source
This commit is contained in:
mightbesimon 2023-06-25 04:47:20 +12:00
parent f13654e6d8
commit cc5fe74d6b
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
core/pathex.py
View file

@ -32,10 +32,10 @@ def get_image_paths(dir_path, image_extensions=image_extensions, subdirs=False,
else: else:
gen = scandir(str(dir_path)) gen = scandir(str(dir_path))
for x in list(gen): for x in gen:
if any([x.name.lower().endswith(ext) for ext in image_extensions]): if any(x.name.lower().endswith(ext) for ext in image_extensions): # listcomp is not needed, any() can handle gencomp
result.append( x.path if not return_Path_class else Path(x.path) ) result.append( Path(x) if return_Path_class else x.path ) # avoid unnecessary negative condition
return sorted(result) return result # scandir should already be sorted
def get_image_unique_filestem_paths(dir_path, verbose_print_func=None): def get_image_unique_filestem_paths(dir_path, verbose_print_func=None):
result = get_image_paths(dir_path) result = get_image_paths(dir_path)