Polyp-Segmentation

1. Introduction

• This repo is a simple research for polyp segmentation task. It is a combination of different great papers and works.
• The idea of this repo is mainly from three papers
  • PraNet: Parallel Reverse Attention Network for Polyp Segmentation
  • Deep Layer Aggregation
  • Attentional Feature Fusion
  • ResNeSt: Split-Attention Networks

2. Framework

Figure 1: Overview of the Net, the image is generated by torchviz

Basically, I just did some minor changes to the PraNet:

• Change the Backbone from ResNet to ResNeSt
• Change feature aggregation to Iterative Feature Aggregation
• Change the Reverse Attention Mechanism to Multi-scale attention feature fusion

3. Citation

@article{fan2020pra,
title={PraNet: Parallel Reverse Attention Network for Polyp Segmentation},
author={Fan, Deng-Ping and Ji, Ge-Peng and Zhou, Tao and Chen, Geng and Fu, Huazhu and Shen, Jianbing and Shao, Ling},
journal={MICCAI},
year={2020}
}


@inproceedings{dai21aff,
title={Attentional Feature Fusion},
author={Yimian Dai and Fabian Gieseke and Stefan Oehmcke and Yiquan Wu and Kobus Barnard},
booktitle={{IEEE} Winter Conference on Applications of Computer Vision, {WACV} 2021}
year={2021}
}

@article{zhang2020resnest,
title={ResNeSt: Split-Attention Networks},
author={Zhang, Hang and Wu, Chongruo and Zhang, Zhongyue and Zhu, Yi and Zhang, Zhi and Lin, Haibin and Sun, Yue and He, Tong and Muller, Jonas and   Manmatha, R. and Li, Mu and Smola, Alexander},
journal={arXiv preprint arXiv:2004.08955},
year={2020}
}

@misc{yu2019deep,
title={Deep Layer Aggregation}, 
author={Fisher Yu and Dequan Wang and Evan Shelhamer and Trevor Darrell},
year={2019},
eprint={1707.06484},
archivePrefix={arXiv},
primaryClass={cs.CV}
}