david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

upd nnlib.py

Browse Source
This commit is contained in:
iperov
2019-02-11 21:29:30 +04:00
parent 6c944d8989
commit d8833e929e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nnlib/nnlib.py
+1 -1
View File
@@ -463,7 +463,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
mask = self.mask mask = self.mask
if self.is_mse: if self.is_mse:
blur_mask = tf_gaussian_blur(max(1, mask.get_shape().as_list()[1] // 32))(mask) blur_mask = tf_gaussian_blur(max(1, mask.get_shape().as_list()[1] // 32))(mask)
return K.mean ( 100*K.square( y_true*blur_mask - y_pred*blur_mask ) ) return K.mean ( 10*K.square( y_true*blur_mask - y_pred*blur_mask ) )
else: else:
return (1.0 - (tf.image.ssim (y_true*mask, y_pred*mask, 1.0))) / 2.0 return (1.0 - (tf.image.ssim (y_true*mask, y_pred*mask, 1.0))) / 2.0
nnlib.DSSIMMSEMaskLoss = DSSIMMSEMaskLoss nnlib.DSSIMMSEMaskLoss = DSSIMMSEMaskLoss