https://github.com/manila95/BASS-Net

Band-Adaptive Spectral-Spatial Feature Learning Deep Neural Network for Hyperspectral Image Classification
https://github.com/manila95/BASS-Net

bass-net computer-vision deep-neural-networks hyperspectral-images landcover-classification

Last synced: 2 months ago
JSON representation

Band-Adaptive Spectral-Spatial Feature Learning Deep Neural Network for Hyperspectral Image Classification

• Host: GitHub
• URL: https://github.com/manila95/BASS-Net
• Owner: manila95
• Created: 2016-11-29T18:36:39.000Z (about 8 years ago)
• Default Branch: master
• Last Pushed: 2019-01-27T00:13:39.000Z (almost 6 years ago)
• Last Synced: 2024-10-28T02:53:07.695Z (3 months ago)
• Topics: bass-net, computer-vision, deep-neural-networks, hyperspectral-images, landcover-classification
• Language: Lua
• Homepage:
• Size: 58.6 KB
• Stars: 70
• Watchers: 11
• Forks: 28
• Open Issues: 1
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• Awesome-Earth-Artificial-Intelligence - BassNet - preprint](https://arxiv.org/abs/1612.00144) - Deep Learning for Land-cover Classification in Hyperspectral Images, (Code)

README

        
# BASS Net

### Band-Adaptive Spectral-Spatial Feature Learning Neural Network for Hyperspectral Image Classification

##### Summary:

Deep learning based landcover classification algorithms have recently been proposed in literature. In hyperspectral images (HSI) they face the challenges of large dimensionality, spatial variability of spectral signatures and scarcity of labeled data. In this work we propose an end-to-end deep learning architecture that extracts band specific spectral-spatial features and performs landcover classification. The architecture has fewer independent connection weights and thus requires lesser number of training data. The method is found to outperform the highest reported accuracies on popular hyperspectral image data sets.

Please refer to our [paper](https://arxiv.org/abs/1612.00144) for details.

![Proposed BASS Net Architecture](Figures/DeLHI_framework_diagram.png)

##### Description of the repository:

This repository contains code to build and test the Configuration 4 architecture in the paper. The code has the option to work on one of the three popular [Hyperspectral Image Classification datasets](http://www.ehu.eus/ccwintco/index.php?title=Hyperspectral_Remote_Sensing_Scenes) viz. Indian Pines, Salinas and Pavia University Scene. 

* Dependencies:

	1. Lua 5.2

	2. Python 2.7

	3. Torch 7

	4. Matio

 	

* Execution instructions:

	1. Run ```preprocessing.py``` for preparing the dataset. 

	

		e.g.

	

		```

		python preprocessing.py --data Indian_pines --patch_size 3

		```

		

	2. Run ```bass-net_model.lua``` for building the model, training and testing.

		e.g.

		

		```

		th bass-net_model.lua --path_dir "./data/" --data Indian_pines --development 1 --optimization "Adam" --learningRate 0.0005 --maxIter 8000 --nbands 10 --patch_size 3 

		```