用于场景图生成的 Graph R-CNN
计算机视觉与模式识别
2018-08-02 v1 机器学习
摘要
我们提出一种称为 Graph R-CNN 的新型场景图生成模型，该模型在检测图像中物体及其关系方面既有效又高效。我们的模型包含一个关系提议网络（RePN），可高效处理图像中物体间二次数量级的潜在关系。我们还提出一种注意力图卷积网络（aGCN），可有效捕获物体与关系间的上下文信息。最后，我们引入一种比现有度量更整体且更真实的新评估指标。我们报告了在现有指标及我们提出指标评估下场景图生成的 SOTA 性能。
引用
@article{arxiv.1808.00191,
title = {Graph R-CNN for Scene Graph Generation},
author = {Jianwei Yang and Jiasen Lu and Stefan Lee and Dhruv Batra and Devi Parikh},
journal= {arXiv preprint arXiv:1808.00191},
year = {2018}
}
备注
16 pages, ECCV 2018 camera ready