PNA in Detectron2
"PNA Progressive Neighborhood Aggregation for Semantic Segmentation Refinement"
In this branch, we provide the code based on Detectron2, and the code based on MMsegmentation is in mmsegmentation branch ( git clone --branch mmsegmentation git@github.com:liutinglt/PNA.git
).
-
Install Detectron2 following the instructions.
-
Install NATTEN, Neighborhood Attention Extension
cd NATTEN
pip install -e .
ln -s your_data_dir ./datasets
Download trained models from google drive
To train a model with 4 GPUs run:
sh train.sh
Testing
sh test.sh
Visualizing
sh demo.sh
If you use PNA, please use the following BibTeX entry.
- PNA:
@inproceedings{liu2023progressive,
title={Progressive Neighborhood Aggregation for Semantic Segmentation Refinement},
author={Liu, Ting and Wei, Yunchao and Zhang, Yanning},
booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
volume={37},
number={2},
pages={1737--1745},
year={2023}
}