Skip to content

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

 
 
 

Contributors