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 cv2
import insightface import insightface
from core.config import get_face from core.config import get_face
@ -9,7 +11,8 @@ FACE_SWAPPER = None
def get_face_swapper(): def get_face_swapper():
global FACE_SWAPPER global FACE_SWAPPER
if FACE_SWAPPER is None: 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 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') quit(f'Python version is not supported - please upgrade to 3.8 or higher')
if not shutil.which('ffmpeg'): if not shutil.which('ffmpeg'):
quit('ffmpeg is not installed!') 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!') quit('File "inswapper_128.onnx" does not exist!')
if '--gpu' in sys.argv: if '--gpu' in sys.argv:
CUDA_VERSION = torch.version.cuda CUDA_VERSION = torch.version.cuda