apps/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-10-05 16:27:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Restore globals, add process time for better comparison

Browse Source
This commit is contained in:
henryruhs
2023-05-30 01:03:43 +02:00
parent 54f800dd0e
commit dc0653ad39
4 changed files with 17 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
core/processor.py
View File

@@ -1,18 +1,20 @@
import os
import time
import cv2
import insightface
import onnxruntime
import core.globals
from core.config import get_face
from core.utils import rreplace
if os.path.isfile('inswapper_128.onnx'):
face_swapper = insightface.model_zoo.get_model('inswapper_128.onnx', providers=onnxruntime.get_available_providers())
face_swapper = insightface.model_zoo.get_model('inswapper_128.onnx', providers=core.globals.providers)
else:
quit('File "inswapper_128.onnx" does not exist!')
def process_video(source_img, frame_paths):
start_time = time.time()
source_face = get_face(cv2.imread(source_img))
for frame_path in frame_paths:
frame = cv2.imread(frame_path)
@@ -28,6 +30,8 @@ def process_video(source_img, frame_paths):
print('E', end='', flush=True)
pass
print(flush=True)
end_time = time.time()
print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)
def process_img(source_img, target_path):