|
|
|
|
|
|
|
|
import numpy as np
|
|
|
import roop.globals
|
|
|
from roop.core import (
|
|
|
start,
|
|
|
decode_execution_providers,
|
|
|
suggest_max_memory,
|
|
|
suggest_execution_threads,
|
|
|
)
|
|
|
from roop.processors.frame.core import get_frame_processors_modules
|
|
|
from roop.utilities import normalize_output_path
|
|
|
from PIL import Image
|
|
|
import gradio as gr
|
|
|
import os
|
|
|
|
|
|
def swap_face(source_file, target_file):
|
|
|
source_image = Image.fromarray(source_file)
|
|
|
source_path = "input.jpg"
|
|
|
source_image.save(source_path, format="JPEG", quality=95)
|
|
|
|
|
|
|
|
|
output_path = "output.mp4"
|
|
|
|
|
|
roop.globals.source_path = source_path
|
|
|
roop.globals.target_path = target_file
|
|
|
|
|
|
|
|
|
|
|
|
roop.globals.output_path = normalize_output_path(
|
|
|
roop.globals.source_path, roop.globals.target_path, output_path
|
|
|
)
|
|
|
roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
|
|
|
roop.globals.headless = True
|
|
|
roop.globals.keep_fps = True
|
|
|
roop.globals.keep_audio = True
|
|
|
roop.globals.keep_frames = False
|
|
|
roop.globals.many_faces = False
|
|
|
roop.globals.video_encoder = "libx264"
|
|
|
roop.globals.video_quality = 50
|
|
|
roop.globals.max_memory = suggest_max_memory()
|
|
|
roop.globals.execution_providers = decode_execution_providers(["cpu"])
|
|
|
roop.globals.execution_threads = suggest_execution_threads()
|
|
|
for frame_processor in get_frame_processors_modules(
|
|
|
roop.globals.frame_processors
|
|
|
):
|
|
|
if not frame_processor.pre_check():
|
|
|
return
|
|
|
|
|
|
start()
|
|
|
|
|
|
return os.path.join(os.getcwd(), output_path)
|
|
|
|
|
|
app = gr.Interface(
|
|
|
fn=swap_face, inputs=[gr.Image(), gr.Video()], outputs=[gr.Video()], description="Video-Face-Swapper [ source : image | target : video ]"
|
|
|
)
|
|
|
|
|
|
app.launch(ssr_mode=False) |
