LSST

This is the official PyTorch implementation of our paper:

Simple and Efficient: A Semisupervised Learning Framework for Remote Sensing Image Semantic Segmentation
Xiaoqiang Lu, Licheng Jiao, Fang Liu, Shuyuan Yang, Xu Liu, Zhixi Feng, Lingling Li, Puhua Chen
Accepted by IEEE Transactions on Geoscience and Remote Sensing (TGRS) 2022

Getting Started

Install

conda create -n lsst python=3.7
pip install -r requirements.txt

Data Preparation

Pre-trained Model

mkdir pretrained
cd pretrained
wget https://download.pytorch.org/models/resnet101-63fe2227.pth
mv resnet101-63fe2227.pth resnet101.pth
cd ..

or download via the following link ResNet-101

Dataset

We have processed the original dataset as mentioned in the paper. You can access the processed dataset directly via the following link.

GID-15 | iSAID | DFC22 | MER | MSL | Vaihingen

File Organization

├── ./pretrained
    └── resnet101.pth
    
├── [Your Dataset Path]
    ├── images
    └── labels

Results

<div style="text-align: center;"> <table> <tr> <td>Dataset</td> <td>Setting</td> <td>Method</td> <td>mIoU</td> </tr> <tr> <td rowspan="4">GID-15</td> <td rowspan="2">1-8</td> <td >baseline</td> <td >61.86</td> </tr> <tr> <td >ours</td> <td >66.38</td> </tr> <tr> <td rowspan="2">1-4</td> <td >baseline</td> <td >67.90</td> </tr> <tr> <td >ours</td> <td >71.28</td> </tr> <tr> <td rowspan="4">iSAID</td> <td rowspan="2">100</td> <td >baseline</td> <td >39.91</td> </tr> <tr> <td >ours</td> <td >46.94</td> </tr> <tr> <td rowspan="2">300</td> <td >baseline</td> <td >60.47</td> </tr> <tr> <td >ours</td> <td >63.40</td> </tr> <tr> <td rowspan="4">DFC22</td> <td rowspan="2">1-8</td> <td >baseline</td> <td >26.97</td> </tr> <tr> <td >ours</td> <td >30.94</td> </tr> <tr> <td rowspan="2">1-4</td> <td >baseline</td> <td >32.67</td> </tr> <tr> <td >ours</td> <td >36.40</td> </tr> <tr> <td rowspan="4">MER</td> <td rowspan="2">1-8</td> <td >baseline</td> <td >43.63</td> </tr> <tr> <td >ours</td> <td >49.68</td> </tr> <tr> <td rowspan="2">1-4</td> <td >baseline</td> <td >48.19</td> </tr> <tr> <td >ours</td> <td >51.31</td> </tr> <tr> <td rowspan="4">MSL</td> <td rowspan="2">1-8</td> <td >baseline</td> <td >50.17</td> </tr> <tr> <td >ours</td> <td >54.72</td> </tr> <tr> <td rowspan="2">1-4</td> <td >baseline</td> <td >50.26</td> </tr> <tr> <td >ours</td> <td >56.22</td> </tr> <tr> <td rowspan="4">Vaihingen</td> <td rowspan="2">1-8</td> <td >baseline</td> <td >53.30</td> </tr> <tr> <td >ours</td> <td >64.09</td> </tr> <tr> <td rowspan="2">1-4</td> <td >baseline</td> <td >59.30</td> </tr> <tr> <td >ours</td> <td >65.34</td> </tr> </table> </div>

Training and Testing

Training

Change DATASET, SPLIT, and DATASET_PATH as you want in train.py, then run:

CUDA_VISIBLE_DEVICES=0,1 python train.py

Testing

Change WEIGHTS, and DATASET_PATH as you want in test.py, then run:

CUDA_VISIBLE_DEVICES=0,1 python test.py

Acknowledgement

The code is mainly inherited from ST++, Thanks a lot for their great works!

Citation

If you find this project useful, please consider citing:

@article{lu2022simple,
  title={Simple and Efficient: A Semisupervised Learning Framework for Remote Sensing Image Semantic Segmentation},
  author={Lu, Xiaoqiang and Jiao, Licheng and Liu, Fang and Yang, Shuyuan and Liu, Xu and Feng, Zhixi and Li, Lingling and Chen, Puhua},
  journal={IEEE Transactions on Geoscience and Remote Sensing},
  volume={60},
  pages={1--16},
  year={2022},
  publisher={IEEE}
}