Skip to content

GaganaMD/Hyperspectral-Wavenet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HYPERSPECTRAL BIORNET - A COMPLETE WAVELET FRAMEWORK FOR UNMIXING

This repository contains the code implementation of our research paper, "Hyperspectral BiorNet: A Complete Wavelet Framework For Unmixing", submitted to IGARSS 2025. The project presents a novel framework leveraging wavelet transformations and neural networks to enhance hyperspectral unmixing accuracy.

Table of Contents

  1. Introduction
  2. Features
  3. Requirements
  4. Installation
  5. Usage
  6. Evaluation
  7. Results
  8. Citation
  9. License and Copyright

Introduction

Hyperspectral unmixing (HSU) is a critical task in remote sensing, involving the decomposition of mixed spectral signatures into pure spectral components (endmembers) and their respective abundances. The Hyperspectral BiorNet leverages:

  • Biorthogonal 3.3 Wavelet Transformations for feature extraction.
  • Wavelet-Inspired Activation Function to enhance spectral-spatial representation.
  • Deep Learning techniques for robust and efficient unmixing.

The framework is designed to address challenges like noise, spectral variability, and overlapping signatures, enabling precise extraction of material components from hyperspectral images.


Features

  • Wavelet Transformation Framework: Biorthogonal 3.3 wavelet decomposition to preserve spatial-spectral features.
  • Wavelet-Inspired Activation Function: Combines oscillatory and smoothing effects for enhanced feature extraction.
  • Hybrid Neural Network: Combines CNN and LSTM layers for endmember extraction and abundance estimation.
  • Evaluation Metrics: Includes RMSE, Spectral Angle Distance (SAD), and Spectral Information Divergence (SID).

Requirements

  • Python 3.8+
  • PyTorch 2.1.1+cu118
  • NumPy
  • SciPy
  • Matplotlib
  • scikit-learn

Installation

Clone the repository:

https://github.com/GaganaMD/Hyperspectral-Wavenet.git
cd Hyperspectral-Wavenet

Install the required dependencies:

pip install -r requirements.txt

Usage

  1. Prepare Data:

    • Ensure your hyperspectral image data is in the correct format.
    • If using synthetic data, preprocess it accordingly. For real-world datasets, extract relevant abundance maps and endmembers.
  2. Run the Notebooks:

    • Data Transformation:

      • Convert your hyperspectral data into the wavelet domain for feature extraction and dimensionality reduction.
      • Apply necessary transformations (e.g., spherical abundance constraints) to optimize the unmixing process.
    • Model Training and Evaluation:

      • Train the wavelet-based model using appropriate transformation techniques.
      • Evaluate the model's performance using standard metrics and methods.
  3. Visualization:

    • Visualize the results, including predicted abundance maps and extracted endmembers.
    • Explore the performance of the custom wavelet-based activation function and its effect on model output.

This general procedure guides you through preparing data, training models, and visualizing results. You can adjust the steps according to the specific dataset or experimental needs. For any questions or clarifications, feel free to reach out.


Evaluation

The model is evaluated using the following metrics:

  1. Root Mean Squared Error (RMSE): Measures prediction error.
  2. Spectral Angle Distance (SAD): Quantifies spectral similarity.
  3. Spectral Information Divergence (SID): Assesses divergence between predicted and ground truth spectra.

Results

Synthetic Data

  • Abundance Maps: Synthetic Abundance Map
  • Extracted Endmembers: Synthetic Endmembers

Real Data (HYDICE Urban Dataset)

  • Abundance Maps: Real Abundance Map
  • Extracted Endmembers: Real Endmembers

Citation

If you use this code or find it helpful, please cite:

@article{Gagana2025HyperspectralBiorNet,
  title={Hyperspectral BiorNet: A Complete Wavelet Framework for Unmixing},
  author={Gagana M D and Ajay Kumar K and Vijayashekhar S S and Jignesh S. Bhatt},
  journal={Proceedings of the Hyperspectral Imaging Workshop},
  year={2025},
  publisher={Acharya Institute of Technology}
}

License and Copyright

This repository contains the official implementation of the research work submitted to IGARSS 2025. The content of the paper and this code are protected under copyright. Citation details will be updated upon acceptance.

For inquiries regarding the code or paper, please contact: [email protected] or [email protected].

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published