From f742640c15e8c44fd53bf3a2d1f7b0aea278e5d1 Mon Sep 17 00:00:00 2001 From: dunnousername <12957182+dunnousername@users.noreply.github.com> Date: Thu, 10 Sep 2020 10:48:18 -0400 Subject: [PATCH] fix some bugs and add start script --- .gitignore | 3 ++- gui.py | 1 + inference.py | 2 +- start.bat | 4 ++++ 4 files changed, 8 insertions(+), 2 deletions(-) create mode 100644 start.bat diff --git a/.gitignore b/.gitignore index cefc15083..e64ab7d53 100644 --- a/.gitignore +++ b/.gitignore @@ -16,4 +16,5 @@ evaluation/test_filelists/lr*.txt *.mp3 wav2lip.pth venv/ -facedetection/detection/sfd/s3fd.pth \ No newline at end of file +facedetection/detection/sfd/s3fd.pth +model/ \ No newline at end of file diff --git a/gui.py b/gui.py index 0c622ea68..9ba3e2a0f 100644 --- a/gui.py +++ b/gui.py @@ -1,3 +1,4 @@ +# -*- coding: utf-8 -*- from tkinter.filedialog import askopenfilename, asksaveasfilename from simpletk import * from inference import main diff --git a/inference.py b/inference.py index b15595bbd..b3c18df1f 100644 --- a/inference.py +++ b/inference.py @@ -12,7 +12,7 @@ parser = argparse.ArgumentParser(description='Inference code to lip-sync videos in the wild using Wav2Lip models') parser.add_argument('--checkpoint_path', type=str, - help='Name of saved checkpoint to load weights from', default='wav2lip.pth') + help='Name of saved checkpoint to load weights from', default='model/wav2lip.pth') parser.add_argument('--face', type=str, help='Filepath of video/image that contains faces to use') diff --git a/start.bat b/start.bat new file mode 100644 index 000000000..36450f24d --- /dev/null +++ b/start.bat @@ -0,0 +1,4 @@ +@echo off +cd /D "%~dp0" +"%~dp0\python.exe" gui.py +pause \ No newline at end of file