- Install pytorch
- The code is implemented on python3.6 and official Pytorch.
- Please install PyTorch-Encoding for Synchronized BN.
- 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
- 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}
}