mhaut/scalable_RNN_HSI

Source code of "Scalable Recurrent Neural Network for Hyperspectral Image Classification"

Scalable Recurrent Neural Network for Hyperspectral Image Classification

The Code for "Scalable Recurrent Neural Network for Hyperspectral Image Classification". [https://link.springer.com/article/10.1007/s11227-020-03187-0]

M. E. Paoletti, J. M. Haut, J. Plaza and A. Plaza.
Scalable Recurrent Neural Network for Hyperspectral Image Classification.
Journal of Supercomputing.
DOI: 10.1007/s11227-020-03187-0
Accepted for publication, 2019.

Example of use

Download datasets

sh retrieveData.sh

Run code

Indian Pines
python -u rnn.py --dataset IP # Proposed
python -u rnn.py --dataset IP --vanillarnn # Vanilla RNN CUDA
python -u rnn.py --dataset IP --vanillarnn --cudnn # Vanilla RNN CuDNN
python -u rnn.py --dataset IP --lstm # LSTM CUDA
python -u rnn.py --dataset IP --lstm --cudnn # LSTM CuDNN
python -u rnn.py --dataset IP --gru # GRU CUDA
python -u rnn.py --dataset IP --gru --cudnn # GRU CuDNN
University of Pavia
python -u rnn.py --dataset PU # Proposed
python -u rnn.py --dataset PU --vanillarnn # Vanilla RNN CUDA
python -u rnn.py --dataset PU --vanillarnn --cudnn # Vanilla RNN CuDNN
python -u rnn.py --dataset PU --lstm # LSTM CUDA
python -u rnn.py --dataset PU --lstm --cudnn # LSTM CuDNN
python -u rnn.py --dataset PU --gru # GRU CUDA
python -u rnn.py --dataset PU --gru --cudnn # GRU CuDNN
Salinas Valley
python -u rnn.py --dataset SV # Proposed
python -u rnn.py --dataset SV --vanillarnn # Vanilla RNN CUDA
python -u rnn.py --dataset SV --vanillarnn --cudnn # Vanilla RNN CuDNN
python -u rnn.py --dataset SV --lstm # LSTM CUDA
python -u rnn.py --dataset SV --lstm --cudnn # LSTM CuDNN
python -u rnn.py --dataset SV --gru # GRU CUDA
python -u rnn.py --dataset SV --gru --cudnn # GRU CuDNN