Skip to content
/ DBFNet Public

[TIP 2022] Deep Bilateral Filtering Network for Point-Supervised Semantic Segmentation in Remote Sensing Images

Notifications You must be signed in to change notification settings

Luffy03/DBFNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

58 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

[TIP 2022] Deep Bilateral Filtering Network for Point-Supervised Semantic Segmentation in Remote Sensing Images

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages