henghuiding/BFP

[ICCV2019] Boundary-Aware Feature Propagation for Scene Segmentation

Boundary-Aware Feature Propagation for Scene Segmentation

1. Install pytorch

• The code is implemented on python3.6 and official Pytorch.
• Please install PyTorch-Encoding for Synchronized BN.

2. Usage

• Training:

python train.py --model  bfp --dataset pascalcontext --checkname bfp101  --batch-size 12

• Testing:

python test.py --model bfp --dataset pascalcontext --resume-dir model_path --eval

3. Citations

• Please consider citing our paper in your publications if the project helps your research.

@inproceedings{ding2019bfp,
  title={Boundary-aware feature propagation for scene segmentation},
  author={Ding, Henghui and Jiang, Xudong and Liu, Ai Qun and Thalmann, Nadia Magnenat and Wang, Gang},
  booktitle={Proceedings of the IEEE International Conference on Computer Vision},
  pages={6819--6829},
  year={2019}
}