https://github.com/haofanwang/kgrn-sr
Official Implementation for Knowledge Graph Routed Network for Situation Recognition [TPAMI'2023]
https://github.com/haofanwang/kgrn-sr
knowledge-graph situation-recognition visual-recognition
Last synced: 12 months ago
JSON representation
Official Implementation for Knowledge Graph Routed Network for Situation Recognition [TPAMI'2023]
- Host: GitHub
- URL: https://github.com/haofanwang/kgrn-sr
- Owner: haofanwang
- Created: 2020-11-18T07:13:27.000Z (over 5 years ago)
- Default Branch: main
- Last Pushed: 2023-01-18T10:08:53.000Z (over 3 years ago)
- Last Synced: 2024-12-27T02:23:08.702Z (over 1 year ago)
- Topics: knowledge-graph, situation-recognition, visual-recognition
- Language: Python
- Homepage:
- Size: 18.9 MB
- Stars: 2
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Knowledge-aware Global Reasoning for Situation Recognition
This repository is the official implementation of KGRN-SR. In this work, we proposes to improve the representation of relationships between semantic concepts by using a global knowledge pool. The paper has been accepted to TPAMI'2023.
## Data Preparation
Download imSitu image set from http://imsitu.org (we use the resized image set)
## Training and Evaluation
All scripts can be found in [./scripts](https://github.com/haofanwang/KGRN-SR/tree/main/scripts).
## Pretrained models
We released pretrained TDA, CAQ, GGNN-baseline, KGRN-R2R, KGRN-R2V, KGRN-R2R-R2V models. You can download [here](https://drive.google.com/drive/folders/1QSgklpRwHfQJxMHckk8n_i9RH3ICt6x9?usp=sharing).
## Contact
For any question, please email to haofanwang.ai@gmail.com