By Chi Wang, Yunke Zhang, Miaomiao Cui, Peiran Ren, Yin Yang, Xuansong Xie, Xian-Sheng Hua, Hujun Bao and Weiwei Xu.
This repo is the official implementation of Active Boundary Loss for Semantic Segmentation
(arXiv:2102.02696) accepted by AAAI 2022.
- Python libarary:
- pytorch>=1.0
- torchvision
- numpy
- scipy
- opencv-python
- label smooth
- [option] Lovász-Softmax loss
- Directly use ABL in
abl.py
. - For ease of use, we provide an all-in-one version. In addition, you can speed it up by moving the process of distance map generation to the dataloader.
If you have any questions, please feel free to contact [email protected]
.