rename dump_dflive to export_dfm

This commit is contained in:
iperov 2021-06-28 18:26:54 +04:00
parent 9d6b6feb1f
commit 63c794b3d0
4 changed files with 12 additions and 12 deletions

10
main.py
View file

@ -148,15 +148,15 @@ if __name__ == "__main__":
p.add_argument('--execute-program', dest="execute_program", default=[], action='append', nargs='+')
p.set_defaults (func=process_train)
def process_dumpdflive(arguments):
def process_exportdfm(arguments):
osex.set_process_lowest_prio()
from mainscripts import DumpDFLive
DumpDFLive.main(model_class_name = arguments.model_name, saved_models_path = Path(arguments.model_dir))
from mainscripts import ExportDFM
ExportDFM.main(model_class_name = arguments.model_name, saved_models_path = Path(arguments.model_dir))
p = subparsers.add_parser( "dumpdflive", help="Dump model to use in DFLive.")
p = subparsers.add_parser( "exportdfm", help="Export model to use in DeepFaceLive.")
p.add_argument('--model-dir', required=True, action=fixPathAction, dest="model_dir", help="Saved models dir.")
p.add_argument('--model', required=True, dest="model_name", choices=pathex.get_all_dir_names_startswith ( Path(__file__).parent / 'models' , 'Model_'), help="Model class name.")
p.set_defaults (func=process_dumpdflive)
p.set_defaults (func=process_exportdfm)
def process_merge(arguments):
osex.set_process_lowest_prio()