pix2pix:该项目使用名为Pix2Pix的条件生成对抗网络(cGAN)进行图像到图像的翻译任务 源码

使用Pix2Pix GAN将Google Satelite Image转换为Streetmap图像 Pix2Pix GAN是图像到图像转换的通用方法。 它基于条件生成对抗网络,其中生成目标图像,该目标图像以给定输入图像为条件。 提出了Pix2Pix GAN的思想。 根据该论文,该模型不仅学习从源图像到目标图像的映射,而且学习损失函数以训练该映射。 网络架构 生成器是经过修改的U-net模型,它将RGB图像作为输入,然后尝试将其映射到相同形状的另一个RGB图像。 鉴别器是一个PatchGan,输出一个30x30的矩阵,然后用于计算对抗损失。 数据集 可以使用此从Kaggle下载数据集。 下载数据集后,将其提取到data / dataset文件夹中。 超参数 source_images = 1096 target_images = 1096 IMAGE_HEIGHT = 256 IMAGE_W