arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1702.07191cs.CV
收藏

ViP-CNN:视觉短语引导的卷积神经网络

ViP-CNN: Visual Phrase Guided Convolutional Neural Network

The Chinese University of Hong Kong(香港中文大学);University of Sydney(悉尼大学);Shenzhen Institutes of Advanced Technology(深圳先进技术研究院)

Yikang Li, Wanli Ouyang, Xiaogang Wang, Xiao'ou Tang

摘要:

视觉关系检测作为连接图像描述和目标检测的中间层任务,以其描述能力强、结构清晰等特点开始受到研究者的关注。它检测对象并捕获它们与主谓宾三元组的成对交互,例如人-骑-马。在本文中,每个视觉关系被认为是一个短语的三个组成部分。我们制定的视觉关系检测为三个相互连接的识别问题,并提出了一个视觉短语引导的卷积神经网络(ViP-CNN),同时解决他们。在ViP-CNN中,我们提出了一个短语引导的消息传递结构(PMPS)来建立关系组件之间的连接,并帮助模型联合考虑这三个问题。提出了相应的非极大值抑制方法和模型训练策略。实验结果表明,我们的ViP-CNN在速度和准确性方面都优于最先进的方法。我们进一步在我们清理的Visual Genome Relationship数据集上预训练ViP-CNN,发现它比ImageNet上的预训练更好。

英文摘要:

As the intermediate level task connecting image captioning and object detection, visual relationship detection started to catch researchers' attention because of its descriptive power and clear structure. It detects the objects and captures their pair-wise interactions with a subject-predicate-object triplet, e.g. person-ride-horse. In this paper, each visual relationship is considered as a phrase with three components. We formulate the visual relationship detection as three inter-connected recognition problems and propose a Visual Phrase guided Convolutional Neural Network (ViP-CNN) to address them simultaneously. In ViP-CNN, we present a Phrase-guided Message Passing Structure (PMPS) to establish the connection among relationship components and help the model consider the three problems jointly. Corresponding non-maximum suppression method and model training strategy are also proposed. Experimental results show that our ViP-CNN outperforms the state-of-art method both in speed and accuracy. We further pretrain ViP-CNN on our cleansed Visual Genome Relationship dataset, which is found to perform better than the pretraining on the ImageNet for this task.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑