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

upd nnlib.py

Browse source
This commit is contained in:
iperov 2019-02-12 09:30:38 +04:00
parent e9313e29f6
commit 429e7e6aee
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
nnlib/nnlib.py
View file

@ -463,7 +463,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
mask = self.mask
if self.is_mse:
blur_mask = tf_gaussian_blur(max(1, mask.get_shape().as_list()[1] // 32))(mask)
return K.mean ( 10*K.square( y_true*blur_mask - y_pred*blur_mask ) )
return K.mean ( 100*K.square( y_true*blur_mask - y_pred*blur_mask ) )
else:
return (1.0 - (tf.image.ssim (y_true*mask, y_pred*mask, 1.0))) / 2.0
nnlib.DSSIMMSEMaskLoss = DSSIMMSEMaskLoss