GAN в Tensorflow2 - черные части сгенерированного изображения - PullRequest
Новая
       77

GAN в Tensorflow2 - черные части сгенерированного изображения

0 голосов
/ 21 февраля 2020

Я пытался создать генератор лиц GAN на основе учебника от tenorflow https://www.tensorflow.org/tutorials/generative/dcgan

Я столкнулся с проблемой сгенерированного изображения. Проблема в том, что выходное изображение повреждено, как вы можете видеть output output generator. Я попытался увеличить уровень шума и количество фильтров на слой, но результат остается прежним. Есть ли способ как это исправить?

В TF2.Keras есть пример кода для архитектуры генератора. 

def make_generator_model(self):
    leak = 0.2
    momentum = 0.95

    model = Sequential()
    model.add(Dense(4*4*1024, input_shape=(self.input_size,)))#, use_bias=False))
    model.add(BatchNormalization(momentum = momentum))
    model.add(LeakyReLU(alpha=leak))

    model.add(Reshape((4, 4, 1024)))

    model.add(Conv2DTranspose(1024, self.kernel_size, strides=(2, 2), padding='same'))
    model.add(BatchNormalization(momentum = momentum))
    model.add(LeakyReLU(alpha=leak))

    model.add(Conv2DTranspose(512, self.kernel_size, strides=(2, 2), padding='same'))
    model.add(BatchNormalization(momentum = momentum))
    model.add(LeakyReLU(alpha=leak))

    model.add(Conv2DTranspose(256, self.kernel_size, strides=(2, 2), padding='same'))
    model.add(BatchNormalization(momentum = momentum))
    model.add(LeakyReLU(alpha=leak))

    model.add(Conv2DTranspose(128, self.kernel_size, strides=(2, 2), padding='same'))
    model.add(BatchNormalization(momentum = momentum))
    model.add(LeakyReLU(alpha=leak))

    model.add(Conv2D(3, self.kernel_size , padding='same', activation='tanh'))

    model.summary()

    return model

Спасибо за любые советы или помощь

...