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

fix loss history

Browse source
This commit is contained in:
iperov 2019-01-11 19:19:29 +04:00
parent e6fa021c3d
commit 2473dabf37
4 changed files with 53 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

21
nnlib/nnlib.py
View file

@ -40,6 +40,7 @@ class nnlib(object):
DSSIMMaskLoss = None
PixelShuffler = None
SubpixelUpscaler = None
AddUniformNoise = None
ResNet = None
UNet = None
@ -101,6 +102,7 @@ DSSIMLoss = nnlib.DSSIMLoss
DSSIMMaskLoss = nnlib.DSSIMMaskLoss
PixelShuffler = nnlib.PixelShuffler
SubpixelUpscaler = nnlib.SubpixelUpscaler
AddUniformNoise = nnlib.AddUniformNoise
"""
code_import_keras_contrib_string = \
"""
@ -541,6 +543,25 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
nnlib.PixelShuffler = PixelShuffler
nnlib.SubpixelUpscaler = PixelShuffler
class AddUniformNoise(keras.layers.Layer):
def __init__(self, power=1.0, minval=-1.0, maxval=1.0, **kwargs):
super(AddUniformNoise, self).__init__(**kwargs)
self.power = power
self.supports_masking = True
self.minval = minval
self.maxval = maxval
def call(self, inputs, training=None):
def noised():
return inputs + self.power*K.random_uniform(shape=K.shape(inputs), minval=self.minval, maxval=self.maxval)
return K.in_train_phase(noised, inputs, training=training)
def get_config(self):
config = {'power': self.power, 'minval': self.minval, 'maxval': self.maxval}
base_config = super(AddUniformNoise, self).get_config()
return dict(list(base_config.items()) + list(config.items()))
nnlib.AddUniformNoise = AddUniformNoise
@staticmethod
def import_keras_contrib(device_config = None):
if nnlib.keras_contrib is not None: