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

fixed model renaming/deleting

Browse source
This commit is contained in:
Colombo 2020-01-26 14:49:37 +04:00
parent c485e1718a
commit 17337ea8f8
2 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
core/pathex.py
View file

@ -52,6 +52,14 @@ def get_image_unique_filestem_paths(dir_path, verbose_print_func=None):
return sorted(result) return sorted(result)
def get_paths(dir_path):
dir_path = Path (dir_path)
if dir_path.exists():
return [ Path(x) for x in sorted([ x.path for x in list(scandir(str(dir_path))) ]) ]
else:
return []
def get_file_paths(dir_path): def get_file_paths(dir_path):
dir_path = Path (dir_path) dir_path = Path (dir_path)

2
models/ModelBase.py
View file

@ -95,7 +95,7 @@ class ModelBase(object):
if is_rename: if is_rename:
new_model_name = io.input_str(f"Enter new name of the model") new_model_name = io.input_str(f"Enter new name of the model")
for filepath in pathex.get_file_paths(saved_models_path): for filepath in pathex.get_paths(saved_models_path):
filepath_name = filepath.name filepath_name = filepath.name
model_filename, remain_filename = filepath_name.split('_', 1) model_filename, remain_filename = filepath_name.split('_', 1)