github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-08-14 02:37:01 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactoring ImageProcessor

Browse source
This commit is contained in:
iperov 2022-05-17 18:30:00 +04:00
commit 2d3d9874bf
2 changed files with 117 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -234,13 +234,12 @@ class FaceMergerWorker(BackendWorker):
face_mask = ImageProcessor(face_mask).erode_blur(state.face_mask_erode, state.face_mask_blur, fade_to_border=True).get_image('HWC')
frame_face_mask = ImageProcessor(face_mask).warp_affine(aligned_to_source_uni_mat, frame_width, frame_height).clip2( (1.0/255.0), 0.0, 1.0, 1.0).get_image('HWC')
face_swap_img = ImageProcessor(face_swap_img).to_ufloat32().get_image('HWC')
face_swap_ip = ImageProcessor(face_swap_img).to_ufloat32()
if state.color_transfer == 'rct':
face_align_img = ImageProcessor(face_align_img).to_ufloat32().get_image('HWC')
face_swap_img = lib_ct.rct(face_swap_img, face_align_img, target_mask=face_mask, source_mask=face_mask)
if state.color_transfer == 'rct':
face_swap_img = face_swap_ip.rct(like=face_align_img, mask=face_mask, like_mask=face_mask)
frame_face_swap_img = ImageProcessor(face_swap_img).warp_affine(aligned_to_source_uni_mat, frame_width, frame_height, interpolation=interpolation).get_image('HWC')
frame_face_swap_img = face_swap_ip.warp_affine(aligned_to_source_uni_mat, frame_width, frame_height, interpolation=interpolation).get_image('HWC')
# Combine final frame
opacity = np.float32(state.face_opacity)