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, <a href="https://scholar.google.com/citations?hl=en&user=Gfa4nasAAAAJ">Leyuan Fang</a>, <a href="https://scholar.google.com/citations?user=epXQ1RwAAAAJ&hl=en&oi=ao">Jun Yue</a>, <a href="https://scholar.google.com/citations?user=dlZuABAAAAAJ&hl=en">Bob Zhang</a>, <a href="https://scholar.google.com/citations?user=Gr9afd0AAAAJ&hl=en">Pedram Ghamisi</a>, and Min He

Getting Started

Prepare Dataset

Download the Potsdam and Vaihingen <b>datasets</b> after processing.

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

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

Evaluate

1. Download the <b>original datasets</b>

2. Download our <b>weights</b>

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.