apps/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2024-08-24 08:31:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

multi-face support

Browse Source
This commit is contained in:
Somdev Sangwan 2023-06-02 14:40:32 +05:30
parent 6a694138de
commit 94e26909aa
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
run.py
View File

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