Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
yrcong committed May 1, 2023
1 parent 9757e1a commit ba0a93d
Showing 1 changed file with 1 addition and 25 deletions.
26 changes: 1 addition & 25 deletions data/README.md
Original file line number Diff line number Diff line change
@@ -1,28 +1,4 @@
:tada: Good News: our work has been accepted by **IEEE Transactions on Pattern Analysis and Machine Intelligence (PAMI)!** :tada:

[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1-U642OoCyb8OSM8nx9lme49dmWa_aUcU?usp=sharing)
# RelTR: Relation Transformer for Scene Graph Generation

We now provide [[Colab](https://colab.research.google.com/drive/1-U642OoCyb8OSM8nx9lme49dmWa_aUcU?usp=sharing)] Demo!

PyTorch Implementation of the Paper [**RelTR: Relation Transformer for Scene Graph Generation**](https://arxiv.org/abs/2201.11460v3)

Different from most existing advanced approaches that infer the **dense** relationships between all entity proposals, our one-stage method can directly generate a **sparse** scene graph by decoding the visual appearance.

<p align="center">
<img src="demo/demo.png">
</p>

# 0. Checklist

- [x] Inference Code :tada:
- [x] Training Code for Visual Genome :tada:
- [x] Evaluation Code for Visual Genome :tada:
- [x] Colab Demo :tada:
- [ ] Training Code for OpenImages V6 :clock9:
- [ ] Evaluation Code for OpenImages V6 :clock9:

# 1. Installation
# 1. For Visual Genome:
Download **RelTR Repo** with:
```
git clone https://github.com/yrcong/RelTR.git
Expand Down

0 comments on commit ba0a93d

Please sign in to comment.