upagge/roop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Resolve absolute model path

Browse Source
This commit is contained in:
henryruhs 2023-05-30 21:12:19 +02:00
parent 47dafb48ba
commit ffc2bd2485
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
core/processor.py
View File

@ -1,3 +1,5 @@
import os
import cv2
import insightface
from core.config import get_face
@ -9,7 +11,8 @@ FACE_SWAPPER = None
def get_face_swapper():
global FACE_SWAPPER
if FACE_SWAPPER is None:
FACE_SWAPPER = insightface.model_zoo.get_model('inswapper_128.onnx')
model_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'inswapper_128.onnx')
FACE_SWAPPER = insightface.model_zoo.get_model(model_path)
return FACE_SWAPPER

3
run.py
View File

@ -64,7 +64,8 @@ def pre_check():
quit(f'Python version is not supported - please upgrade to 3.8 or higher')
if not shutil.which('ffmpeg'):
quit('ffmpeg is not installed!')
if os.path.isfile('../inswapper_128.onnx'):
model_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'inswapper_128.onnx')
if not os.path.isfile(model_path):
quit('File "inswapper_128.onnx" does not exist!')
if '--gpu' in sys.argv:
CUDA_VERSION = torch.version.cuda