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

ConverterMasked: better lct

Browse source
This commit is contained in:
iperov 2019-02-14 18:43:24 +04:00
parent fbf6b5f25e
commit 75eb7468ee
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/ConverterMasked.py
View file

@ -220,7 +220,8 @@ class ConverterMasked(ConverterBase):
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) ] 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) ]
prd_face_bgr = image_utils.reinhard_color_transfer ( np.clip( (prd_face_bgr*255).astype(np.uint8), 0, 255), prd_face_bgr = image_utils.reinhard_color_transfer ( np.clip( (prd_face_bgr*255).astype(np.uint8), 0, 255),
np.clip( (dst_face_bgr*255).astype(np.uint8), 0, 255) ) np.clip( (dst_face_bgr*255).astype(np.uint8), 0, 255),
source_mask=prd_face_mask_a, target_mask=prd_face_mask_a)
prd_face_bgr = np.clip( prd_face_bgr.astype(np.float32) / 255.0, 0.0, 1.0) prd_face_bgr = np.clip( prd_face_bgr.astype(np.float32) / 255.0, 0.0, 1.0)
if debug: if debug:

4
samples/SampleGeneratorFace.py
View file

@ -157,10 +157,10 @@ class SampleGeneratorFace(SampleGeneratorBase):
pitch, yaw = LandmarksProcessor.estimate_pitch_yaw (sample.landmarks) pitch, yaw = LandmarksProcessor.estimate_pitch_yaw (sample.landmarks)
if self.add_pitch: if self.add_pitch:
batches[i_pitch].append (pitch) batches[i_pitch].append ([pitch])
if self.add_yaw: if self.add_yaw:
batches[i_yaw].append (yaw) batches[i_yaw].append ([yaw])
break break
yield [ np.array(batch) for batch in batches] yield [ np.array(batch) for batch in batches]