文件名称:pix2pix:该项目使用名为Pix2Pix的条件生成对抗网络(cGAN)进行图像到图像的翻译任务
文件大小:5.14MB
文件格式:ZIP
更新时间:2024-04-14 09:33:01
computer-vision deep-learning tensorflow gan cyclegan
使用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
【文件预览】:
pix2pix-main
----images()
--------result2.png(508KB)
--------discriminator.png(275KB)
--------result.png(554KB)
----data()
--------preprocessed_data()
--------dataset()
----models()
--------download_model.txt(154B)
----notebooks()
--------EDA.ipynb(2.15MB)
--------restore_model.ipynb(482KB)
--------pix2pix.ipynb(213KB)
----README.md(2KB)
----preprocess_data.py(2KB)
----test_images()
--------4.jpg(124KB)
--------3.jpg(135KB)
--------1.jpg(129KB)
--------2.jpg(111KB)
--------5.jpg(103KB)