github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-08-19 04:59:28 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix

Browse source
This commit is contained in:
iperov 2021-10-02 00:13:29 +04:00
commit 9e50b59f6f
2 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
apps/DeepFaceLive/backend/FaceMerger.py
View file

@ -196,10 +196,10 @@ class FaceMergerWorker(BackendWorker):
if state.face_mask_type == FaceMaskType.SRC:
face_mask_t = lib_cl.Tensor.from_value(face_align_mask_img)
face_mask_t = face_mask_t.transpose( (2,0,1), op_text='O = (I <= 128 ? 0 : 1);', dtype=np.uint8)
face_mask_t = face_mask_t.transpose( (2,0,1), op_text='O = (I < 128 ? 0 : 1);', dtype=np.uint8)
elif state.face_mask_type == FaceMaskType.CELEB:
face_mask_t = lib_cl.Tensor.from_value(face_swap_mask_img)
face_mask_t = face_mask_t.transpose( (2,0,1), op_text='O = (I <= 128 ? 0 : 1);', dtype=np.uint8)
face_mask_t = face_mask_t.transpose( (2,0,1), op_text='O = (I < 128 ? 0 : 1);', dtype=np.uint8)
elif state.face_mask_type == FaceMaskType.SRC_M_CELEB:
face_mask_t = lib_cl.any_wise('float X = (((float)I0) / 255.0) * (((float)I1) / 255.0); O = (X <= 0.5 ? 0 : 1);',

4
xlib/avecl/_internal/op/binary_morph.py
View file

@ -3,7 +3,7 @@ from .binary_dilate_circle import binary_dilate_circle
from .binary_erode_circle import binary_erode_circle
from .gaussian_blur import gaussian_blur
from .pad import pad
from .cast import cast
def binary_morph(input_t : Tensor, erode_dilate : int, blur : float, fade_to_border : bool = False, dtype=None) -> Tensor:
"""
@ -39,5 +39,7 @@ def binary_morph(input_t : Tensor, erode_dilate : int, blur : float, fade_to_bor
if blur > 0:
x = gaussian_blur(x, blur * 0.250, dtype=dtype)
else:
x = cast(x, dtype=dtype)
return x[...,H:-H,W:-W]