tensorflow2_pix2pix

1.主体部分
详细代码见我的github
数据可从http://cmp.felk.cvut.cz/~tylecr1/facade/上下载
下载完后放到dataset里面即可
导入相应的包

import numpy as np
from tensorflow.keras import Model
from tensorflow.keras.initializers import RandomNormal, Zeros
from tensorflow.keras.layers import Input, ZeroPadding2D, Conv2D, BatchNormalization, Activation, Dropout, Add, Conv2DTranspose, \
    LeakyReLU, Concatenate
debug = True
image_source_dir = './dataset/facades/'
direction = 'b2a'
input_channel = 3  # input image channels
output_channel = 3  # output image channels
lr = 0.0002
epoch = 150
crop_from = 286
image_size = 256
batch_size = 4
combined_filepath = 'best_weights.h5'
generator_filepath = 'generator.h5'
seed = 9584
imagenet_mean = np.array([0.5, 0.5, 0.5])
imagenet_std = np.array([0.5, 0.5, 0.5])

1.生成鉴别器discriminator

def get_discriminator(n_layers=4, use_sigmoid=True):
    input = Input(shape=(ima