Skip to content

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

License

Notifications You must be signed in to change notification settings

mhaut/scalable_RNN_HSI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

ROhsi

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

About

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

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published