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

refactoring

Browse source
This commit is contained in:
iperov 2021-10-20 18:05:56 +04:00
parent 7463515bfc
commit 78d80f9c5c
5 changed files with 135 additions and 67 deletions
Download patch file Download diff file Expand all files Collapse all files

13
main.py
View file

@ -52,6 +52,19 @@ def main():
p.add_argument('--delete-parts', action="store_true", default=False)
p.set_defaults(func=run_merge_large_files)
train_parser = subparsers.add_parser( "train", help="Train neural network.")
train_parsers = train_parser.add_subparsers()
def train_FaceAligner(args):
faceset_path = Path(args.faceset_path)
from apps.trainers.FaceAligner.FaceAlignerTrainer import FaceAlignerTrainer
FaceAlignerTrainer(faceset_path=faceset_path).run()
p = train_parsers.add_parser('FaceAligner')
p.add_argument('--faceset-path', default=None, action=fixPathAction, help=".dfs path")
p.set_defaults(func=train_FaceAligner)
def bad_args(arguments):
parser.print_help()
exit(0)