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

fix avatar multigpu mode

Browse source
This commit is contained in:
iperov 2018-06-07 12:18:42 +04:00
commit 58d9b4ae09
2 changed files with 37 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

13
nnlib/__init__.py
View file

@ -43,6 +43,19 @@ def DSSIMMaskLossClass(tf):
return total_loss
return DSSIMMaskLoss
def DSSIMLossClass(tf):
class DSSIMLoss(object):
def __init__(self, is_tanh=False):
self.is_tanh = is_tanh
def __call__(self,y_true, y_pred):
if not self.is_tanh:
return (1.0 - tf.image.ssim (y_true, y_pred, 1.0)) / 2.0
else:
return (1.0 - tf.image.ssim ((y_true/2+0.5), (y_pred/2+0.5), 1.0)) / 2.0
return DSSIMLoss
def MSEMaskLossClass(keras):
class MSEMaskLoss(object):