AttentionGAN

收录理由

AttentionGAN 是官方 PyTorch 实现，针对无配对图像到图像转换任务，在 CycleGAN 风格生成器中引入可学习的前景与背景注意力掩码。它并非一次性整体翻译整张图，而是让模型自行判断输出中哪些区域应保留、输入背景中哪些部分需维持，这在自拍到动漫、马到斑马、地图到航拍等转换任务上效果尤为直观。仓库提供预训练模型、数据集下载脚本以及训练和测试命令，既能复现论文结果，也方便将代码迁移到自己的域对。此外还包含多域变体，可用于面部属性迁移，以及面向几何形变的版本，适合涉及形状变化的转换场景。需要留意的是，代码采用非商业学术许可发布，若团队计划商用，应事先联系作者获取授权。