cd models/obj_simi/
python build.py build_ext --inplace
We finetune the I2C
model on the ILSVRC dataset.
cd scripts
sh train.sh
Use the test script to generate attention maps.
cd scripts
sh val.sh
If you find this code helpful, please consider to cite this paper:
@inproceedings{zhang2020inter,
title={Inter-Image Communication for Weakly Supervised Localization},
author={Zhang, Xiaolin and Wei, Yunchao and Yang, Yi},
booktitle={European Conference on Computer Vision},
year={2020},
organization={Springer}
}