Deep Bilateral Filtering Network (DBFNet)

Code for TIP 2022 paper, "Deep Bilateral Filtering Network for Point-Supervised Semantic Segmentation in Remote Sensing Images", accepted.

Authors: Linshan Wu, Leyuan Fang, Jun Yue, Bob Zhang, Pedram Ghamisi, and Min He

Getting Started

Prepare Dataset

Download the Potsdam and Vaihingen datasets after processing.

Or you can download the datasets from the official website. Then, crop the original images and create point labels following our code in Dataprocess.

If your want to run our code on your own datasets, the pre-process code is also available in Dataprocess.

Evaluate

1. Download the original datasets

2. Download our weights

3. Run our code

python predict.py

Train

1. Train DBFNet

python run/point/p_train.py

2. Generate pseudo labels

python run/point/p_predict_train.py

3. Recursive learning

python run/second/sec_train.py

Citation ✏️ 📄

If you find this repo useful for your research, please consider citing the paper as follows:

@ARTICLE{Wu_DBFNet,
  author={Wu, Linshan and Fang, Leyuan and Yue, Jun and Zhang, Bob and Ghamisi, Pedram and He, Min},
  journal={IEEE Transactions on Image Processing}, 
  title={Deep Bilateral Filtering Network for Point-Supervised Semantic Segmentation in Remote Sensing Images}, 
  year={2022},
  volume={31},
  number={},
  pages={7419-7434},
  doi={10.1109/TIP.2022.3222904}}
@article{wu2024modeling,
  title={Modeling the Label Distributions for Weakly-Supervised Semantic Segmentation},
  author={Wu, Linshan and Zhong, Zhun and Ma, Jiayi and Wei, Yunchao and Chen, Hao and Fang, Leyuan and Li, Shutao},
  journal={arXiv preprint arXiv:2403.13225},
  year={2024}
}
@inproceedings{AGMM,
  title={Sparsely Annotated Semantic Segmentation with Adaptive Gaussian Mixtures},
  author={Wu, Linshan and Zhong, Zhun and Fang, Leyuan and He, Xingxin and Liu, Qiang and Ma, Jiayi and Chen, Hao},
  booktitle={IEEE Conf. Comput. Vis. Pattern Recog.},
  month={June},
  year={2023},
  pages={15454-15464}
  }

For any question, please contact Linshan Wu.