graph-rcnn.pytorch

> 编程语言

开源

[ECCV 2018] "Graph R-CNN for Scene Graph Generation" 的官方代码

748 stars0 点赞1 次浏览

访问官网 GitHub

工具介绍

[ECCV 2018] "Graph R-CNN for Scene Graph Generation" 的官方代码

graph-rcnn.pytorch

我们的 ECCV 2018 论文 "Graph R-CNN for Scene Graph Generation" 的 Pytorch 代码

简介

此项目是一组基于 Pytorch 1.0 重新实现的代表性场景图生成模型,包括:

Graph R-CNN for Scene Graph Generation, 我们自己的论文。ECCV 2018。
通过迭代消息传递生成场景图, Xu 等人, CVPR 2017
从对象、短语和区域说明生成场景图, Li 等人, ICCV 2017
神经模式: 利用全局上下文解析场景图, Zellers 等人, CVPR 2018
图形对比损失用于场景图生成, Zhang 等人, CVPR 2019

我们的重新实现基于以下仓库:

将所有这些代表性方法汇集到一个仓库的目的是在相同设置下建立更公平的比较。正如您在最近的文献中注意到的，对于 IMP、MSDN、Graph R-CNN 和 Neural Motifs 的报告数字通常令人困惑，特别是由于 IMP 风格方法（前三种）和 Neural Motifs 风格方法（神经模式论文和基于其的其他变体）之间存在巨大差距。我们希望这个仓库能够为各种场景图生成方法建立一个良好的基准，并为研究社区做出贡献！