Andras Fekete
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from refacer import Refacer
|
|
from os.path import exists
|
|
import argparse
|
|
import cv2
|
|
|
|
parser = argparse.ArgumentParser(description='Refacer')
|
|
parser.add_argument("--force_cpu", help="Force CPU mode", default=False, action="store_true")
|
|
parser.add_argument("--colab_performance", help="Use in colab for better performance", default=False,action="store_true")
|
|
parser.add_argument("--face", help="Face to replace (ex: <src>,<dst>,<thresh=0.2>)", nargs='+', action="append", required=True)
|
|
parser.add_argument("--video", help="Video to parse", required=True)
|
|
args = parser.parse_args()
|
|
|
|
refacer = Refacer(force_cpu=args.force_cpu,colab_performance=args.colab_performance)
|
|
|
|
def run(video_path,faces):
|
|
video_path_exists = exists(video_path)
|
|
if video_path_exists == False:
|
|
print ("Can't find " + video_path)
|
|
return
|
|
|
|
faces_out = []
|
|
for face in faces:
|
|
face_str = face[0].split(",")
|
|
origin = exists(face_str[0])
|
|
if origin == False:
|
|
print ("Can't find " + face_str[0])
|
|
return
|
|
destination = exists(face_str[1])
|
|
if destination == False:
|
|
print ("Can't find " + face_str[1])
|
|
return
|
|
|
|
faces_out.append({
|
|
'origin':cv2.imread(face_str[0]),
|
|
'destination':cv2.imread(face_str[1]),
|
|
'threshold':float(face_str[2])
|
|
})
|
|
|
|
return refacer.reface(video_path,faces_out)
|
|
|
|
run(args.video, args.face)
|