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

Make deep swapper inputs universal

Browse Source
This commit is contained in:
henryruhs 2024-11-14 13:11:24 +01:00
parent 95bcf67a75
commit ca035068fe
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -140,10 +140,10 @@ def forward(crop_vision_frame : VisionFrame) -> Tuple[VisionFrame, Mask, Mask]:
deep_swapper = get_inference_pool().get('deep_swapper') deep_swapper = get_inference_pool().get('deep_swapper')
deep_swapper_inputs = {} deep_swapper_inputs = {}
for deep_swapper_input in deep_swapper.get_inputs(): for index, deep_swapper_input in enumerate(deep_swapper.get_inputs()):
if deep_swapper_input.name == 'in_face:0': if index == 0:
deep_swapper_inputs[deep_swapper_input.name] = crop_vision_frame deep_swapper_inputs[deep_swapper_input.name] = crop_vision_frame
if deep_swapper_input.name == 'morph_value:0': if index == 1:
morph_value = numpy.array([ 1 ]).astype(numpy.float32) morph_value = numpy.array([ 1 ]).astype(numpy.float32)
deep_swapper_inputs[deep_swapper_input.name] = morph_value deep_swapper_inputs[deep_swapper_input.name] = morph_value