Giter Club home page Giter Club logo

tgrs_2021_mdn-asp's Introduction

Multi-direction Networks with Attentional Spectral Prior for Hyperspectral Image Classification, TGRS, 2021.

Bobo Xi, Jiaojiao Li, Yunsong Li, Rui song, Yuchao Xiao, Yanzi Shi and Qian Du.


Code for paper: Multi-direction Networks with Attentional Spectral Prior for Hyperspectral Image Classification.

Fig. 1: The framework of our proposed MDN-ASP for HSI classification. It is composed of four components: multi-direction samples construction, multi-stream feature extraction, feature aggregation with attentional spectral prior (ASP) and a softmax-based classifier. The same color represents the layers with same operation.

Training and Test Process

Please firstly run the 'generate_train_val_test_gt.m' to generate the training and test maps. Then, run 'construct_multi_mat.py' to construct the multi-direction samples. Finally, run 'main_MDN_ASP.py' to reproduce the MDN-ASP results on IndianPines data set. The training samples distribution and the obtained classification map are shown below. We have successfully test it on Ubuntu 16.04 and Windows systems with Matlab R2017b. Part of the source code are employed from the works of DR-CNN and SSRN.

Fig. 2: The composite false-color image, groundtruth, training samples, and classification map of Indian Pines dataset.

References

If you find this code helpful, please kindly cite:

[1] B. Xi, J. Li, Y. Li, R. Song, Y. Xiao, Y. Shi, Q. Du "Multi-direction Networks with Attentional Spectral Prior for Hyperspectral Image Classification," in IEEE Transactions on Geoscience and Remote Sensing, 2021, doi: 10.1109/TGRS.2020.3047682
[2] B. Xi, J. Li, Y. Li, R. Song, Y. Shi, S. Liu, Q. Du "Deep Prototypical Networks With Hybrid Residual Attention for Hyperspectral Image Classification," in IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, vol. 13, pp. 3683-3700, 2020, doi: 10.1109/JSTARS.2020.3004973.

Citation Details

BibTeX entry:

@ARTICLE{Xi_TGRS2020_MDNASP,
  author={B. {Xi} and J. {Li} and Y. {Li} and R. {Song} and Y. {Xiao} and Y. {Shi} and Q. {Du}},
  journal={IEEE Transactions on Geoscience and Remote Sensing}, 
  title={Multi-Direction Networks With Attentional Spectral Prior for Hyperspectral Image Classification}, 
  year={2021},
  volume={},
  number={},
  pages={1-15},
  doi={10.1109/TGRS.2020.3047682}}
@ARTICLE{Xi2020JSTARS,
  author={B. {Xi} and J. {Li} and Y. {Li} and R. {Song} and Y. {Shi} and S. {Liu} and Q. {Du}},
  journal={IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing}, 
  title={Deep Prototypical Networks With Hybrid Residual Attention for Hyperspectral Image Classification}, 
  year={2020},
  volume={13},
  number={},
  pages={3683-3700},}

Licensing

Copyright (C) 2020 Bobo Xi

This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 3 of the License.

This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with this program.

tgrs_2021_mdn-asp's People

Contributors

b-xi avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.