4 个月前

通过对抗训练从模拟图像和无监督图像中学习

通过对抗训练从模拟图像和无监督图像中学习

摘要

随着图形技术的最新进展,使用合成图像训练模型变得更为可行,这可能避免了昂贵的标注需求。然而,由于合成图像与真实图像分布之间存在差距,从合成图像中学习可能无法达到预期的性能。为了缩小这一差距,我们提出了模拟+无监督(Simulated+Unsupervised, S+U)学习方法,其目标是在保留模拟器提供的标注信息的同时,利用未标注的真实数据来提高模拟器输出的逼真度。我们开发了一种用于S+U学习的方法,该方法使用类似于生成对抗网络(Generative Adversarial Networks, GANs)的对抗网络,但输入为合成图像而非随机向量。为了保留标注信息、避免伪影并稳定训练过程,我们在标准GAN算法中进行了几项关键修改:(i) 自正则化项(self-regularization term),(ii) 局部对抗损失(local adversarial loss),以及 (iii) 使用历史改进图像更新判别器。我们展示了这种方法能够生成高度逼真的图像,并通过定性和用户研究进行了验证。我们通过训练用于注视估计和手势估计的模型对生成的图像进行了定量评估。结果显示,与仅使用合成图像相比,我们的方法显著提高了性能,并在无需任何真实数据标注的情况下,在MPIIGaze数据集上达到了最先进的结果。

代码仓库

rickyhan/SimGAN-Captcha
tf
GitHub 中提及
adnanalam53/cycleGAN
tf
GitHub 中提及
rvorias/uvHolographics
GitHub 中提及
mjdietzx/SimGAN
tf
GitHub 中提及
ajdillhoff/simgan-pytorch
pytorch
GitHub 中提及
AlexHex7/SimGAN_pytorch
pytorch
GitHub 中提及
ashkanpakzad/atn
pytorch
GitHub 中提及

基准测试

基准方法指标
image-to-image-translation-on-cityscapesSimGAN
Class IOU: 0.04
Per-class Accuracy: 10%
Per-pixel Accuracy: 20%
image-to-image-translation-on-cityscapes-1SimGAN
Class IOU: 0.07
Per-class Accuracy: 11%
Per-pixel Accuracy: 47%

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
通过对抗训练从模拟图像和无监督图像中学习 | 论文 | HyperAI超神经