# ML_GCN.pytorch
PyTorch implementation of [Multi-Label Image Recognition with Graph Convolutional Networks](https://arxiv.org/abs/1904.03582), CVPR 2019.
### Requirements
Please, install the following packages
- numpy
- torch-0.3.1
- torchnet
- torchvision-0.2.0
- tqdm
### Download pretrain models
checkpoint/coco ([GoogleDrive](https://drive.google.com/open?id=1ivLi1Rc-dCUmN1ProcMk76zxF1DSvlIk))
checkpoint/voc ([GoogleDrive](https://drive.google.com/open?id=1lhbmW5g-Mo9KgI07nmc1kwSbEnb6t-YA))
or
[Baidu](https://pan.baidu.com/s/17j3lTjMRmXvWHT86zhaaVA)
### Options
- `lr`: learning rate
- `lrp`: factor for learning rate of pretrained layers. The learning rate of the pretrained layers is `lr * lrp`
- `batch-size`: number of images per batch
- `image-size`: size of the image
- `epochs`: number of training epochs
- `evaluate`: evaluate model on validation set
- `resume`: path to checkpoint
### Demo VOC 2007
```sh
python3 demo_voc2007_gcn.py data/voc --image-size 448 --batch-size 32 -e --resume checkpoint/voc/voc_checkpoint.pth.tar
```
### Demo COCO 2014
```sh
python3 demo_coco_gcn.py data/coco --image-size 448 --batch-size 32 -e --resume checkpoint/coco/coco_checkpoint.pth.tar
```
## Citing this repository
If you find this code useful in your research, please consider citing us:
```
@inproceedings{ML_GCN_CVPR_2019,
author = {Zhao-Min, Chen and Xiu-Shen, Wei and Peng, Wang and Yanwen, Guo},
title = {{Multi-Label Image Recognition with Graph Convolutional Networks}},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
year = {2019}
}
```
## Reference
This project is based on https://github.com/durandtibo/wildcat.pytorch
## Tips
If you have any questions about our work, please do not hesitate to contact us by emails.
没有合适的资源?快使用搜索试试~ 我知道了~
资源详情
资源评论
资源推荐
收起资源包目录
ML_GCN.zip (18个子文件)
ML_GCN-master
util.py 11KB
engine.py 19KB
.DS_Store 6KB
demo_voc2007_gcn.py 3KB
demo_coco_gcn.py 4KB
coco.py 5KB
__init__.py 0B
models.py 3KB
README.md 2KB
voc.py 9KB
data
voc
voc_glove_word2vec.pkl 24KB
voc_adj.pkl 3KB
.DS_Store 6KB
coco
coco_glove_word2vec.pkl 94KB
coco_adj.pkl 51KB
checkpoint
voc
.DS_Store 6KB
.DS_Store 6KB
coco
.DS_Store 6KB
共 18 条
- 1
余淏
- 粉丝: 56
- 资源: 3973
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功
评论2