Home

Awesome

Pointly-Supervised Panoptic Segmentation (ECCV 2022 Oral)

Panoptic segmentation with a single point label per target! [paper]

Here are all the codes to reproduce the results. We are still cleaning them and will update later.

The code is based on the implementation of Panoptic SegFormer.

The point labels used in the paper can be downloaded here.

Install

  1. Install common prerequisties: Python, PyTorch, CUDA, etc.
  2. Install mmcv. This work is tested with mmcv up to version 1.7.1.
  3. Install other requirements:
    pip install -r requirements.txt
    
  4. Install the custom pydijkstra package:
    cd PSPS/py-dijkstra/pydijkstra
    python setup.py install --user
    

Note that the panopticapi has risk of memory leakage, as been discussed in this issue. The solution is to add a workers.close() in panopticapi/evaluation.py/pq_compute_multi_core before the function return.

Prepare Dataset

Train & Test

Please find the example_run.sh for details.

TODO