fi0r1/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-06 18:08:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

conditional process according to --frame-processor

Browse Source
This commit is contained in:
Moeblack 2023-06-10 17:16:52 +08:00 committed by henryruhs
parent dc2282faf0
commit f278f92085
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
roop/core.py
View File

@ -181,7 +181,8 @@ def start() -> None:
if has_image_extention(roop.globals.target_path):
if predict_image(roop.globals.target_path) > 0.85:
destroy()
roop.swapper.process_image(roop.globals.source_path, roop.globals.target_path, roop.globals.output_path)
if 'face-swapper' in roop.globals.frame_processor:
roop.swapper.process_image(roop.globals.source_path, roop.globals.target_path, roop.globals.output_path)
if roop.globals.gpu_vendor == 'nvidia' and 'face-enhancer' in roop.globals.frame_processor:
roop.enhancer.process_image(roop.globals.source_path, roop.globals.target_path, roop.globals.output_path)
elif 'face-enhancer' in roop.globals.frame_processor: