face-swap / app.py
shuhulhandoo's picture
Update app.py
0972d04
raw
history blame contribute delete
908 Bytes
#! /usr/bin/env python
import os
import cv2
import argparse
from face_detection import select_face, select_all_faces
from face_swap import face_swap
import gradio as gr
from PIL import Image
def swap(src_img, dst_img):
# src_img = cv2.imread(src_img.name)
# dst_img = cv2.imread(Image.open(dst_img))
src_points, src_shape, src_face = select_face(src_img)
dst_faceBoxes = select_all_faces(dst_img)
if dst_faceBoxes is None:
print('Detect 0 Face !!')
exit(-1)
output = dst_img
for k, dst_face in dst_faceBoxes.items():
output = face_swap(src_face, dst_face["face"], src_points,
dst_face["points"], dst_face["shape"],
output, "correct color")
return output
if __name__ == '__main__':
# swap()
demo = gr.Interface(fn=swap, inputs=["image", "image"], outputs="image")
demo.launch()