diff --git a/README.md b/README.md index 58c40f2..6f4fc3f 100644 --- a/README.md +++ b/README.md @@ -1,5 +1,8 @@ +[![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.11460) 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. @@ -13,6 +16,7 @@ Different from most existing advanced approaches that infer the **dense** relati - [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: