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.生成鉴别器discriminat