You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
A pytorch based implementation of Hybrid-Spectral-Net as in IEEE GRSL paper "HybridSN: Exploring 3D-2D CNN Feature Hierarchy for Hyperspectral Image Classification".
Hybrid-Spectral-Net for Hyperspectral Image Classification.
Description
The HybridSN is spectral-spatial 3D-CNN followed by spatial 2D-CNN. The 3D-CNN facilitates the joint spatial-spectral feature representation from a stack of spectral bands. The 2D-CNN on top of the 3D-CNN further learns more abstract level spatial representation.
Paper link: https://ieeexplore.ieee.org/document/8736016 .
Original Keras implimentation: https://github.com/gokriznastic/HybridSN
Model
Fig: HybridSpectralNet (HybridSN) Model with 3D and 2D convolutions for hyperspectral image (HSI) classification.
Requirements
To install requirements:
conda env create -f environment.yml
To download the dataset and setup the folders, run:
bash setup_script.sh
Training
To train the model(s) in the paper, run this command in the A2S2KResNet folder:
A pytorch based implementation of Hybrid-Spectral-Net as in IEEE GRSL paper "HybridSN: Exploring 3D-2D CNN Feature Hierarchy for Hyperspectral Image Classification".