An extended Python implementation of COLA-NET
COLA-Net: Collaborative Attention Network for Image Restoration
This repository is for COLA-Net introduced in the following paper: Chong Mou, Jian Zhang, Xiaopeng Fan, Hangfan Liu, and Ronggang Wang, "COLA-Net: Collaborative Attention Network for Image Restoration", (IEEE Transactions on Multimedia 2021)
Citation
If you find their code helpful in your resarch or work, please cite the following papers.
@inproceedings{zhang2019rnan,
title={Residual Non-local Attention Networks for Image Restoration},
author={Zhang, Yulun and Li, Kunpeng and Li, Kai and Zhong, Bineng and Fu, Yun},
booktitle={ICLR},
year={2019}
}
@article{mou2021cola,
title={COLA-Net: Collaborative Attention Network for Image Restoration},
author={Chong, Mou and Jian, Zhang and Xiaopeng, Fan and Hangfan, Liu and Ronggang, Wang},
journal={IEEE Transactions on Multimedia},
year={2021}
}
Acknowledgements
This code is built on RNAN (PyTorch). We thank the authors for sharing their codes of RNAN.