Semantic Segmentation for Remote Sensing

This repo is the PyTorch implementation of some works related to remote sensing tasks.

Methods

SAM-based:

SAM_RS (TGRS Undergoing Review).

Mamba-based:

RS^3Mamba (GRSL Undergoing Review).

Unsupervised Domain Adaptation (UDA):

GLGAN (TGRS Undergoing Review).

MBATA_GAN (IEEE TGRS 2023).

Multimodal Fusion:

FTransUNet (IEEE TGRS 2024).

CMFNet (IEEE JSTARS 2022).

MSFNet (IEEE IGARSS 2022).

Reference

For UDA:

• TransGAN
• Advent
• AdaSegNet

For Semantic Segmentation:

• TransUNet
• UNetformer
• UCTransNet
• Swin-Transformer
• Swin-Unet
• v-FuseNet
• FuseNet

For Semantic Segmentation based on Mamba:

• VMamba
• VM-UNet
• Swin-UMamba

Datasets

All datasets including ISPRS Potsdam, ISPRS Vaihingen, loveDA can be downloaded here.

Other Works

There are some other works in our group:

Scene Classification: DF4LCZ (IEEE TGRS Undergoing Review).

Change Detection: GCD-DDPM (IEEE TGRS 2024), GVLM (ISPRS 2023).

Super-resolution: ASDDPM (IEEE JSTARS Undergoing Review), GCRDN (IEEE JSTARS 2023).

Citations

If these codes are helpful for your study, please cite:

@article{ma2024rs3mamba,
  title={RS3Mamba: Visual State Space Model for Remote Sensing Images Semantic Segmentation},
  author={Ma, Xianping and Zhang, Xiaokang and Pun, Man-On},
  journal={arXiv preprint arXiv:2404.02457},
  year={2024}
}

@ARTICLE{ma2024ftransunet,
  author={Ma, Xianping and Zhang, Xiaokang and Pun, Man-On and Liu, Ming},
  journal={IEEE Transactions on Geoscience and Remote Sensing}, 
  title={A Multilevel Multimodal Fusion Transformer for Remote Sensing Semantic Segmentation}, 
  year={2024},
  volume={62},
  number={},
  pages={1-15},
  doi={10.1109/TGRS.2024.3373033}
}

@article{ma2023sam,
  title={SAM-Assisted Remote Sensing Imagery Semantic Segmentation with Object and Boundary Constraints},
  author={Ma, Xianping and Wu, Qianqian and Zhao, Xingyu and Zhang, Xiaokang and Pun, Man-On and Huang, Bo},
  journal={arXiv preprint arXiv:2312.02464},
  year={2023}
}

@article{ma2023unsupervised,
  title={Unsupervised domain adaptation augmented by mutually boosted attention for semantic segmentation of vhr remote sensing images},
  author={Ma, Xianping and Zhang, Xiaokang and Wang, Zhiguo and Pun, Man-On},
  journal={IEEE Transactions on Geoscience and Remote Sensing},
  volume={61},
  pages={1--15},
  year={2023},
  publisher={IEEE}
}

@article{ma2022crossmodal,
  title={A crossmodal multiscale fusion network for semantic segmentation of remote sensing data},
  author={Ma, Xianping and Zhang, Xiaokang and Pun, Man-On},
  journal={IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing},
  volume={15},
  pages={3463--3474},
  year={2022},
  publisher={IEEE}
}

@inproceedings{ma2022msfnet,
  title={MSFNET: Multi-Stage Fusion Network for Semantic Segmentation of Fine-Resolution Remote Sensing Data},
  author={Ma, Xianping and Zhang, Xiaokang and Pun, Man-On and Liu, Ming},
  booktitle={IGARSS 2022-2022 IEEE International Geoscience and Remote Sensing Symposium},
  pages={2833--2836},
  year={2022},
  organization={IEEE}
}

Contact

Xianping Ma (xianpingma@ling.cuhk.edu.cn), (ma.xianping125@gmail.com)

Star History