File size: 2,285 Bytes
2c98260 881953f 2c98260 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 | import cv2
import gradio as gr
import numpy as np
from tinyface import TinyFace
# Initialize TinyFace once at startup
tinyface = TinyFace()
tinyface.prepare()
def swap_face(original_image: np.ndarray, reference_image: np.ndarray) -> np.ndarray:
# Gradio provides RGB numpy arrays; TinyFace/OpenCV expects BGR
original_bgr = cv2.cvtColor(original_image, cv2.COLOR_RGB2BGR)
reference_bgr = cv2.cvtColor(reference_image, cv2.COLOR_RGB2BGR)
# Detect the face in the original image (used to identify which face to replace)
source_face = tinyface.get_one_face(original_bgr)
if source_face is None:
raise gr.Error("No face detected in the original image.")
# Detect the new face to apply (from the reference image)
new_face = tinyface.get_one_face(reference_bgr)
if new_face is None:
raise gr.Error("No face detected in the reference image.")
# swap_face(image, reference_face=face to match, destination_face=new face to apply)
result_bgr = tinyface.swap_face(original_bgr, source_face, new_face)
# Convert back to RGB for Gradio
return cv2.cvtColor(result_bgr, cv2.COLOR_BGR2RGB)
with gr.Blocks(title="Face Swap") as demo:
gr.Markdown("# Face Swap")
gr.Markdown(
"Upload an original image and a reference image. "
"The face in the original image will be replaced with the face from the reference image."
)
with gr.Row():
with gr.Column():
original = gr.Image(label="Original Image (face to replace)", type="numpy")
reference = gr.Image(label="Reference Image (new face to apply)", type="numpy")
with gr.Row():
clear_btn = gr.ClearButton([original, reference], value="Clear")
submit_btn = gr.Button("Submit", variant="primary")
with gr.Column():
result = gr.Image(label="Result")
submit_btn.click(fn=swap_face, inputs=[original, reference], outputs=result)
clear_btn.add(result)
gr.Markdown(
'<div style="text-align:center; margin-top:1rem; color:#888; font-size:0.85rem;">'
'Powered by <a href="https://www.tertiarycourses.com.sg/" target="_blank">'
"Tertiary Infotech Academy Pte Ltd</a></div>"
)
if __name__ == "__main__":
demo.launch()
|