github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 21:12:07 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

_

Browse source
This commit is contained in:
Colombo 2020-01-31 09:41:54 +04:00
parent 3e63ba2d17
commit 126c2bce5c
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
merger/MergeMasked.py
View file

@ -45,7 +45,7 @@ def MergeMaskedFace (predictor_func, predictor_input_shape, cfg, frame_info, img
prd_face_mask_a_0 = cv2.resize (dst_face_mask_a_0, predictor_input_shape[0:2] ) prd_face_mask_a_0 = cv2.resize (dst_face_mask_a_0, predictor_input_shape[0:2] )
predictor_masked = False predictor_masked = False
if cfg.super_resolution_mode: if cfg.super_resolution_mode != 0:
prd_face_bgr = cfg.superres_func(cfg.super_resolution_mode, prd_face_bgr) prd_face_bgr = cfg.superres_func(cfg.super_resolution_mode, prd_face_bgr)
prd_face_bgr = np.clip(prd_face_bgr, 0, 1) prd_face_bgr = np.clip(prd_face_bgr, 0, 1)
@ -269,7 +269,7 @@ def MergeMaskedFace (predictor_func, predictor_input_shape, cfg, frame_info, img
k_size = int(frame_info.motion_power*cfg_mp) k_size = int(frame_info.motion_power*cfg_mp)
if k_size >= 1: if k_size >= 1:
k_size = np.clip (k_size+1, 2, 50) k_size = np.clip (k_size+1, 2, 50)
if cfg.super_resolution_mode: if cfg.super_resolution_mode != 0:
k_size *= 2 k_size *= 2
out_face_bgr = imagelib.LinearMotionBlur (out_face_bgr, k_size , frame_info.motion_deg) out_face_bgr = imagelib.LinearMotionBlur (out_face_bgr, k_size , frame_info.motion_deg)