PEBAL

This repo contains the Pytorch implementation of our paper:

Pixel-wise Energy-biased Abstention Learning for Anomaly Segmentation on Complex Urban Driving Scenes

Yu Tian*, Yuyuan Liu*, Guansong Pang, Fengbei Liu, Yuanhong Chen, Gustavo Carneiro.

• SOTA on 4 benchmarks. Check out Papers With Code.

Inference

Checkpoint for anomaly segmentation

After downloading the pre-trained checkpoint, simply run the following command:

python test.py

Training Code will be released soon

Citation

If you find this repo useful for your research, please consider citing our paper:

@misc{tian2021pixelwise,
      title={Pixel-wise Energy-biased Abstention Learning for Anomaly Segmentation on Complex Urban Driving Scenes}, 
      author={Yu Tian and Yuyuan Liu and Guansong Pang and Fengbei Liu and Yuanhong Chen and Gustavo Carneiro},
      year={2021},
      eprint={2111.12264},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}