Skip to content

liutinglt/PNA

Repository files navigation

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).

Installation

  1. Install Detectron2 following the instructions.

  2. Install NATTEN, Neighborhood Attention Extension

cd NATTEN
pip install -e .

Prepare the datasets

 ln -s your_data_dir ./datasets

Download trained models from google drive

Training & Testing & Visualizing

We have provided the corresponding scripts, just modify it accordingly.

To train a model with 4 GPUs run:

sh train.sh

Testing

sh test.sh

Visualizing

sh demo.sh

Citing PNA

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}
} 

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published