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 wangchi1995@zju.edu.cn
.