| from typing import Any, List, Callable |
| import roop.globals |
| import insightface |
| import cv2 |
| import numpy as np |
|
|
| from roop.typing import Face, Frame |
| from roop.utilities import resolve_relative_path |
|
|
|
|
|
|
| class FaceSwapInsightFace(): |
| model_swap_insightface = None |
|
|
|
|
| processorname = 'faceswap' |
| type = 'swap' |
|
|
|
|
| def Initialize(self, devicename): |
| if self.model_swap_insightface is None: |
| model_path = resolve_relative_path('../models/inswapper_128.onnx') |
| self.model_swap_insightface = insightface.model_zoo.get_model(model_path, providers=roop.globals.execution_providers) |
|
|
| |
| def Run(self, source_face: Face, target_face: Face, temp_frame: Frame) -> Frame: |
| img_fake, M = self.model_swap_insightface.get(temp_frame, target_face, source_face, paste_back=False) |
| target_face.matrix = M |
| return img_fake |
|
|
|
|
| def Release(self): |
| del self.model_swap_insightface |
| self.model_swap_insightface = None |
|
|
|
|
| |
|
|
|
|
|
|
|
|
