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

nothing interesting

Browse source
This commit is contained in:
iperov 2019-02-28 20:10:58 +04:00
parent 46f4d6c426
commit 7cea93c0f2
1 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
nnlib/nnlib.py
View file

@ -166,6 +166,9 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
if 'TF_SUPPRESS_STD' in os.environ.keys() and os.environ['TF_SUPPRESS_STD'] == '1': if 'TF_SUPPRESS_STD' in os.environ.keys() and os.environ['TF_SUPPRESS_STD'] == '1':
suppressor = std_utils.suppress_stdout_stderr().__enter__() suppressor = std_utils.suppress_stdout_stderr().__enter__()
#if "tensorflow" in device_config.backend:
# nnlib.keras = nnlib.tf.keras
#else:
import keras as keras_ import keras as keras_
nnlib.keras = keras_ nnlib.keras = keras_
@ -318,13 +321,12 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
nnlib.dssim = dssim nnlib.dssim = dssim
class PixelShuffler(keras.layers.Layer): class PixelShuffler(keras.layers.Layer):
def __init__(self, size=(2, 2), data_format=None, **kwargs): def __init__(self, size=(2, 2), data_format='channels_last', **kwargs):
super(PixelShuffler, self).__init__(**kwargs) super(PixelShuffler, self).__init__(**kwargs)
self.data_format = K.normalize_data_format(data_format) self.data_format = data_format
self.size = keras.utils.conv_utils.normalize_tuple(size, 2, 'size') self.size = size
def call(self, inputs): def call(self, inputs):
input_shape = K.int_shape(inputs) input_shape = K.int_shape(inputs)
if len(input_shape) != 4: if len(input_shape) != 4:
raise ValueError('Inputs should have rank ' + raise ValueError('Inputs should have rank ' +