DiscoGAN

DiscoGAN

PyTorch 官方实现 [学习发现跨域关系] 使用生成对抗网络](https://arxiv.org/pdf/1703.05192.pdf)。 前提条件

• Python 2.7
• PyTorch
• Numpy/Scipy/Pandas
• 进度条
• OpenCV 训练 DiscoGAN

CelebA

使用以下方法下载 CelebA 数据集: $ Python ./datasets/download.py celebA (目前,下载 CelebA 数据集的链接已不可用)。 为了训练性别转换, $ Python ./discogan/image_translation.py --task_name='celebA' --style_A='Male' 为训练头发颜色转换

$ Python ./discogan/image_translation.py --task_name='celebA' --style_A='Blond_Hair' --style_B='Black_Hair' --constraint='Male'

手提包/鞋子

使用以下方法下载 Edges2Handbags 数据集: $ Python ./datasets/download.py edges2handbags 使用以下方法下载 Edges2Shoes 数据集: $ Python ./datasets/download.py edges2shoes 为了训练 Edges2Handbags, $ Python ./discogan/image_translation.py --task_name='edges2handbags' 为了训练 Edges2Shoes, $ Python ./discogan/image_translation.py --task_name='edges2shoes' 要训练 Handbags2Shoes, $ Python ./discogan/image_translation.py --task_name='Handbags2Shoes' --starting_rate=0.5

面部磨砂

使用以下方法下载 Facescrub 数据集

$ Python ./datasets/download.py facescrub 要训练性别转换，请运行：$ Python ./discogan/image_translation.py --task_name='facescrub' 从 3D car dataset 和 3D face dataset 中下载数据，并解压到 ./datasets 目录中。要训练 Car2Car 翻译，请运行：$ Python ./discogan/angle_pairing.py --task_name='car2car' 要训练 Car2Face 翻译，请运行：$ Python ./discogan/angle_pairing.py --task_name='car2face' 运行 script.sh 脚本，以使用其他数据集训练模型，请在相应行中取消注释。结果