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

Fix bug in ConverterMasked when overexposed pixels appear sometimes

Browse source
This commit is contained in:
iperov 2019-01-09 13:57:26 +04:00
parent faa8cebc98
commit 1e28ae24b1
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/ConverterMasked.py
View file

@ -256,6 +256,8 @@ class ConverterMasked(ConverterBase):
if self.alpha: if self.alpha:
out_img = np.concatenate ( [out_img, np.expand_dims (img_mask_blurry_aaa[:,:,0],-1)], -1 ) out_img = np.concatenate ( [out_img, np.expand_dims (img_mask_blurry_aaa[:,:,0],-1)], -1 )
out_img = np.clip (out_img, 0.0, 1.0 )
if debug: if debug:
debugs += [out_img.copy()] debugs += [out_img.copy()]