* Pytorch 0.3.1
* Python 3.6
Please enter the main folder, and run
python3 demo_coco__adapt.py --data data/coco --image-size 448 --batch-size 32 -e --resume checkpoint/coco/coco_checkpoint.pth.tar
The pre-trained model on coco at GoogleDrive
The pre-trained model on fashion550k at GoogleDrive
If you find our work helpful in your research, please kindly cite our paper:
Li Q, Peng X, Qiao Y, et al. Learning Category Correlations for Multi-label Image Recognition with Graph Networks[J]. arXiv preprint arXiv:1909.13005, 2019.
bib:
@article{li2019learning,
title={Learning Category Correlations for Multi-label Image Recognition with Graph Networks},
author={Li, Qing and Peng, Xiaojiang and Qiao, Yu and Peng, Qiang},
journal={arXiv preprint arXiv:1909.13005},
year={2019}
}
This project is based on https://github.com/Megvii-Nanjing/ML_GCN