mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Show/hide morph slider for deep swapper (#822)

Browse Source
This commit is contained in:
Henry Ruhs 2024-11-18 10:59:40 +01:00 committed by GitHub
parent aada3ff618
commit 48016eaba3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 26 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
facefusion/processors/modules/deep_swapper.py
View File

@ -182,10 +182,10 @@ def forward(crop_vision_frame : VisionFrame, deep_swapper_morph : DeepSwapperMor
deep_swapper = get_inference_pool().get('deep_swapper') deep_swapper = get_inference_pool().get('deep_swapper')
deep_swapper_inputs = {} deep_swapper_inputs = {}
for index, deep_swapper_input in enumerate(deep_swapper.get_inputs()): for deep_swapper_input in deep_swapper.get_inputs():
if index == 0: if deep_swapper_input.name == 'in_face:0':
deep_swapper_inputs[deep_swapper_input.name] = crop_vision_frame deep_swapper_inputs[deep_swapper_input.name] = crop_vision_frame
if index == 1: if deep_swapper_input.name == 'morph_value:0':
deep_swapper_inputs[deep_swapper_input.name] = deep_swapper_morph deep_swapper_inputs[deep_swapper_input.name] = deep_swapper_morph
with thread_semaphore(): with thread_semaphore():
@ -194,6 +194,15 @@ def forward(crop_vision_frame : VisionFrame, deep_swapper_morph : DeepSwapperMor
return crop_vision_frame[0], crop_source_mask[0], crop_target_mask[0] return crop_vision_frame[0], crop_source_mask[0], crop_target_mask[0]
def has_morph_input() -> bool:
deep_swapper = get_inference_pool().get('deep_swapper')
for deep_swapper_input in deep_swapper.get_inputs():
if deep_swapper_input.name == 'morph_value:0':
return True
return False
def prepare_crop_frame(crop_vision_frame : VisionFrame) -> VisionFrame: def prepare_crop_frame(crop_vision_frame : VisionFrame) -> VisionFrame:
crop_vision_frame = cv2.addWeighted(crop_vision_frame, 1.75, cv2.GaussianBlur(crop_vision_frame, (0, 0), 2), -0.75, 0) crop_vision_frame = cv2.addWeighted(crop_vision_frame, 1.75, cv2.GaussianBlur(crop_vision_frame, (0, 0), 2), -0.75, 0)
crop_vision_frame = crop_vision_frame / 255.0 crop_vision_frame = crop_vision_frame / 255.0

26
facefusion/uis/components/deep_swapper_options.py
View File

@ -6,6 +6,7 @@ from facefusion import state_manager, wording
from facefusion.common_helper import calc_int_step from facefusion.common_helper import calc_int_step
from facefusion.processors import choices as processors_choices from facefusion.processors import choices as processors_choices
from facefusion.processors.core import load_processor_module from facefusion.processors.core import load_processor_module
from facefusion.processors.modules.deep_swapper import has_morph_input
from facefusion.processors.typing import DeepSwapperModel from facefusion.processors.typing import DeepSwapperModel
from facefusion.uis.core import get_ui_component, register_ui_component from facefusion.uis.core import get_ui_component, register_ui_component
@ -17,26 +18,27 @@ def render() -> None:
global DEEP_SWAPPER_MODEL_DROPDOWN global DEEP_SWAPPER_MODEL_DROPDOWN
global DEEP_SWAPPER_MORPH_SLIDER global DEEP_SWAPPER_MORPH_SLIDER
has_deep_swapper = 'deep_swapper' in state_manager.get_item('processors')
DEEP_SWAPPER_MODEL_DROPDOWN = gradio.Dropdown( DEEP_SWAPPER_MODEL_DROPDOWN = gradio.Dropdown(
label = wording.get('uis.deep_swapper_model_dropdown'), label = wording.get('uis.deep_swapper_model_dropdown'),
choices = processors_choices.deep_swapper_models, choices = processors_choices.deep_swapper_models,
value = state_manager.get_item('deep_swapper_model'), value = state_manager.get_item('deep_swapper_model'),
visible = 'deep_swapper' in state_manager.get_item('processors') visible = has_deep_swapper
) )
DEEP_SWAPPER_MORPH_SLIDER = gradio.Slider( DEEP_SWAPPER_MORPH_SLIDER = gradio.Slider(
label=wording.get('uis.deep_swapper_morph_slider'), label = wording.get('uis.deep_swapper_morph_slider'),
value=state_manager.get_item('deep_swapper_morph'), value = state_manager.get_item('deep_swapper_morph'),
step=calc_int_step(processors_choices.deep_swapper_morph_range), step = calc_int_step(processors_choices.deep_swapper_morph_range),
minimum=processors_choices.deep_swapper_morph_range[0], minimum = processors_choices.deep_swapper_morph_range[0],
maximum=processors_choices.deep_swapper_morph_range[-1], maximum = processors_choices.deep_swapper_morph_range[-1],
visible='deep_swapper' in state_manager.get_item('processors') visible = has_deep_swapper and has_morph_input()
) )
register_ui_component('deep_swapper_model_dropdown', DEEP_SWAPPER_MODEL_DROPDOWN) register_ui_component('deep_swapper_model_dropdown', DEEP_SWAPPER_MODEL_DROPDOWN)
register_ui_component('deep_swapper_morph_slider', DEEP_SWAPPER_MORPH_SLIDER) register_ui_component('deep_swapper_morph_slider', DEEP_SWAPPER_MORPH_SLIDER)
def listen() -> None: def listen() -> None:
DEEP_SWAPPER_MODEL_DROPDOWN.change(update_deep_swapper_model, inputs = DEEP_SWAPPER_MODEL_DROPDOWN, outputs = DEEP_SWAPPER_MODEL_DROPDOWN) DEEP_SWAPPER_MODEL_DROPDOWN.change(update_deep_swapper_model, inputs = DEEP_SWAPPER_MODEL_DROPDOWN, outputs = [ DEEP_SWAPPER_MODEL_DROPDOWN, DEEP_SWAPPER_MORPH_SLIDER ])
DEEP_SWAPPER_MORPH_SLIDER.release(update_deep_swapper_morph, inputs = DEEP_SWAPPER_MORPH_SLIDER) DEEP_SWAPPER_MORPH_SLIDER.release(update_deep_swapper_morph, inputs = DEEP_SWAPPER_MORPH_SLIDER)
processors_checkbox_group = get_ui_component('processors_checkbox_group') processors_checkbox_group = get_ui_component('processors_checkbox_group')
@ -46,17 +48,17 @@ def listen() -> None:
def remote_update(processors : List[str]) -> Tuple[gradio.Dropdown, gradio.Slider]: def remote_update(processors : List[str]) -> Tuple[gradio.Dropdown, gradio.Slider]:
has_deep_swapper = 'deep_swapper' in processors has_deep_swapper = 'deep_swapper' in processors
return gradio.Dropdown(visible = has_deep_swapper), gradio.Slider(visible = has_deep_swapper) return gradio.Dropdown(visible = has_deep_swapper), gradio.Slider(visible = has_deep_swapper and has_morph_input())
def update_deep_swapper_model(deep_swapper_model : DeepSwapperModel) -> gradio.Dropdown: def update_deep_swapper_model(deep_swapper_model : DeepSwapperModel) -> Tuple[gradio.Dropdown, gradio.Slider]:
deep_swapper_module = load_processor_module('deep_swapper') deep_swapper_module = load_processor_module('deep_swapper')
deep_swapper_module.clear_inference_pool() deep_swapper_module.clear_inference_pool()
state_manager.set_item('deep_swapper_model', deep_swapper_model) state_manager.set_item('deep_swapper_model', deep_swapper_model)
if deep_swapper_module.pre_check(): if deep_swapper_module.pre_check():
return gradio.Dropdown(value = state_manager.get_item('deep_swapper_model')) return gradio.Dropdown(value = state_manager.get_item('deep_swapper_model')), gradio.Slider(visible = has_morph_input())
return gradio.Dropdown() return gradio.Dropdown(), gradio.Slider()
def update_deep_swapper_morph(deep_swapper_morph : int) -> None: def update_deep_swapper_morph(deep_swapper_morph : int) -> None: