Skip to content

Latest commit

 

History

History
54 lines (38 loc) · 1.89 KB

README.md

File metadata and controls

54 lines (38 loc) · 1.89 KB
Raw

ADD-GCN: Attention-Driven Dynamic Graph Convolutional Network for Multi-Label Image Recognition

This project hosts the code for implementing the ADD-GCN algorithm for multi-label image recognition, as presented in our paper:

Attention-Driven Dynamic Graph Convolutional Network for Multi-Label Image Recognition;
Jin Ye, Junjun He, Xiaojiang Peng, Wenhao Wu, Yu Qiao;
In: European Conference on Computer Vision (ECCV), 2020.
arXiv preprint arXiv:2012.02994

The full paper is available at: https://arxiv.org/abs/2012.02994.

Installation

This project is implemented with Pytorch and has been tested on version Pytorch 1.0/1.1/1.2.

A quick demo

After you have installed Pytorch, you can follow the below steps to run a quick demo.

Inference for COCO2014

python main.py --data COCO2014 --data_root_dir {YOUR-ROOT-DATA-DIR} --model_name ADD_GCN --resume {THE-TEST-MODEL} -e -i 448

Please note that:

  1. You should put the COCO2014 folder in {YOUR-ROOT-DATA-DIR}.

  2. You should put the test model in {THE-TEST-MODEL} folder.

  3. You can get the same ADD-GCN results with this model. The password is 4ebj.

Model Test size mAP
ResNet-101 448×448 79.7
DecoupleNet 448×448 82.2
ML-GCN 448×448 83.0
ADD-GCN 448×448 84.2
ResNet-101 576×576 80.0
SSGRL 576×576 84.2
ML-GCN 576×576 84.3
ADD-GCN 576×576 85.2

Citations

Please consider citing our paper in your publications if the project helps your research. BibTeX reference is as follows.

@inproceedings{ye2020add,
  title   =  {Attention-Driven Dynamic Graph Convolutional Network for Multi-Label Image Recognition},
  author  =  {Jin Ye, Junjun He, Xiaojiang Peng, Wenhao Wu, Yu Qiao},
  booktitle =  {European Conference on Computer Vision (ECCV)},
  year    =  {2020}
}

License