| 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) | |