From 94e26909aa8d6e48be02cd8724c76eff2a425638 Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Fri, 2 Jun 2023 14:40:32 +0530 Subject: [PATCH] multi-face support --- run.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/run.py b/run.py index 6acf785..6428ee8 100755 --- a/run.py +++ b/run.py @@ -87,8 +87,6 @@ def pre_check(): quit(f"CUDNN version {CUDNN_VERSION} is not supported - please downgrade to 8.9.1") else: core.globals.providers = ['CPUExecutionProvider'] - if '--all-faces' in sys.argv or '-a' in sys.argv: - core.globals.all_faces = True def start_processing(): @@ -193,17 +191,19 @@ def start(): print("\n[WARNING] No face detected in source image. Please try with another one.\n") return if is_img(target_path): - if predict_image(target_path) > 0.7: + if predict_image(target_path) > 0.85: quit() process_img(args['source_img'], target_path, args['output_file']) status("swap successful!") return seconds, probabilities = predict_video_frames(video_path=args['target_path'], frame_interval=100) - if any(probability > 0.7 for probability in probabilities): + if any(probability > 0.85 for probability in probabilities): quit() video_name_full = target_path.split("/")[-1] video_name = os.path.splitext(video_name_full)[0] output_dir = os.path.dirname(target_path) + "/" + video_name + if output_dir.startswith("/"): + output_dir = "." + output_dir Path(output_dir).mkdir(exist_ok=True) status("detecting video's FPS...") fps, exact_fps = detect_fps(target_path)