4 个月前

基于场景图的图像生成

基于场景图的图像生成

摘要

为了真正理解视觉世界,我们的模型不仅应该能够识别图像,还应该能够生成图像。为此,最近在从自然语言描述生成图像方面取得了令人兴奋的进展。这些方法在有限的领域内(如鸟类或花卉的描述)取得了惊人的效果,但在忠实再现包含多个对象和复杂关系的句子时仍面临挑战。为克服这一局限性,我们提出了一种基于场景图生成图像的方法,该方法能够显式地推理对象及其关系。我们的模型利用图卷积来处理输入图,通过预测对象的边界框和分割掩码来计算场景布局,并使用级联细化网络将布局转换为图像。网络通过对抗一对判别器进行训练,以确保输出的真实感。我们在Visual Genome和COCO-Stuff数据集上验证了我们的方法,定性和消融实验以及用户研究均表明,我们的方法能够生成包含多个对象的复杂图像。

代码仓库

roeiherz/CanonicalSg2Im
pytorch
GitHub 中提及
google/sg2im
pytorch
GitHub 中提及
cjpurackal/sg2im-tf
tf
GitHub 中提及
roeiherz/AG2Video
pytorch
GitHub 中提及

基准测试

基准方法指标
layout-to-image-generation-on-coco-stuff-2SG2Im
FID: 67.96
Inception Score: 7.3
layout-to-image-generation-on-visual-genome-2SG2Im
FID: 74.61
Inception Score: 6.3

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于场景图的图像生成 | 论文 | HyperAI超神经