Spectral-Spatial Unified Networks for Hyperspectral Image Classification

Keras implementation of our method for hyperspectral image classification.

Paper

Spectral–Spatial Unified Networks for Hyperspectral Image Classification

Please cite our papers if you find it useful for your research.

@article{ssun,
  author={Y. Xu and L. Zhang and B. Du and F. Zhang},
  journal={IEEE Trans. Geos. Remote Sens.},
  title={Spectral-Spatial Unified Networks for Hyperspectral Image Classification},
  year={2018},
  volume={56},
  number={10},
  pages={5893-5909},
  ISSN={0196-2892},
  month={Oct}
}

@inproceedings{bglstm,
  title={A Band Grouping Based LSTM Algorithm for Hyperspectral Image Classification},
  author={Y. Xu and B. Du and L. Zhang and F. Zhang},
  booktitle={CCF Chinese Conference on Computer Vision},
  pages={421--432},
  year={2017},
  organization={Springer}
}

Installation

Install Keras 2.2.4 from https://github.com/keras-team/keras with Python 3.6.
- Note: This repo is now updated with the Tensorflow backend engine. We have tested the code with Tensorflow 1.13. For the Theano backend users, please refer to https://keras.io/#configuring-your-keras-backend for technical support.
Clone this repo.

git clone https://github.com/YonghaoXu/SSUN

Dataset

Download the Pavia University image and the corresponding annotations.

Usage

Replace the file path for the hyperspectral data in HyperFunctions.py with yours.
Run SSUN.py.
Change the s1s2 index in SSUN.py to switch from different grouping strategies.
- Left: Strategy 1 s1s2 = 1
- Right: Strategy 2 s1s2 = 2

Note

12/2019: Update the code with the Tensorflow backend engine.