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

Fix indentations

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-14 12:28:12 -07:00
commit 7d864776e8
1 changed files with 13 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

26
converters/ConverterMasked.py
View file

@ -366,21 +366,21 @@ class ConverterMasked(Converter):
ColorTransferMode.MASKED_RCT_PAPER_CLIP: (True, True, True),
}
use_masks, use_paper, use_clip = ct_options[self.color_transfer_mode]
use_masks, use_paper, use_clip = ct_options[self.color_transfer_mode]
if not use_masks:
img_bgr = imagelib.reinhard_color_transfer(prd_face_bgr, dst_face_bgr, clip=use_clip,
preserve_paper=use_paper)
else:
img_bgr = imagelib.reinhard_color_transfer(prd_face_bgr, dst_face_bgr, clip=use_clip,
preserve_paper=use_paper, source_mask=prd_face_mask_a,
target_mask=prd_face_mask_a)
if not use_masks:
img_bgr = imagelib.reinhard_color_transfer(prd_face_bgr, dst_face_bgr, clip=use_clip,
preserve_paper=use_paper)
else:
img_bgr = imagelib.reinhard_color_transfer(prd_face_bgr, dst_face_bgr, clip=use_clip,
preserve_paper=use_paper, source_mask=prd_face_mask_a,
target_mask=prd_face_mask_a)
if debug:
debugs += [np.clip(cv2.warpAffine(prd_face_bgr, face_output_mat, img_size,
np.zeros(img_bgr.shape, dtype=np.float32),
cv2.WARP_INVERSE_MAP | cv2.INTER_LANCZOS4,
cv2.BORDER_TRANSPARENT), 0, 1.0)]
if debug:
debugs += [np.clip(cv2.warpAffine(prd_face_bgr, face_output_mat, img_size,
np.zeros(img_bgr.shape, dtype=np.float32),
cv2.WARP_INVERSE_MAP | cv2.INTER_LANCZOS4,
cv2.BORDER_TRANSPARENT), 0, 1.0)]
if self.mode == 'hist-match-bw':
prd_face_bgr = cv2.cvtColor(prd_face_bgr, cv2.COLOR_BGR2GRAY)