Skip to content

Hyperspectal Image classification using state of the art generative adverserial network

Notifications You must be signed in to change notification settings

jaikanthjay46/HSI-Classification-GAN

Repository files navigation

HSI-Classification-GAN

An implementation of HSI Classification using GANs There are two kinds of GAN implemented, 1D GAN uses only spectral features, and 3D GAN uses both spatial and spectral features. To run the code locally, install the required dependencies (Keras, scikit-learn, patchify, numpy, pandas),

Steps to run locally

1D-GAN

Train Phase

python3 1dgan_train.py

Test Phase

python3 1dgan_test.py

3D-GAN

Train Phase

python3 3dgan_train.py

Test Phase

python3 3dgan_test.py

The code uses the pretrained models to predict by default, you should be able to change the model used in the test script.

Steps to run on Google Colab

  • Download and import the ipynb file in Colab
  • Add the indianpines-hyperspectral.zip dataset zip file to the drive root.
  • Also, add the pretrained models to models/ folder in your drive.

About

Hyperspectal Image classification using state of the art generative adverserial network

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published