fix some bugs and add start script

dunnousername · Sep 10, 2020 · f742640 · f742640
1 parent 6e08765
commit f742640
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -16,4 +16,5 @@ evaluation/test_filelists/lr*.txt
 *.mp3
 wav2lip.pth
 venv/
-facedetection/detection/sfd/s3fd.pth
+facedetection/detection/sfd/s3fd.pth
+model/
diff --git 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
@@ -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
@@ -0,0 +1,4 @@
+@echo off
+cd /D "%~dp0"
+"%~dp0\python.exe" gui.py
+pause