Giter VIP home page Giter VIP logo

wfcg's Introduction

Weighted Feature Fusion of Convolutional Neural Network and Graph Attention Network for Hyperspectral Image Classification

This example implements the paper in review [Weighted Feature Fusion of Convolutional Neural Network and Graph Attention Network for Hyperspectral Image Classification]

Run

If you want to run this code, just put your data in the Datasets folder and change a few paths.

  • path 1: datasets: Please put the corresponding hyperspectral data there.
  • path 2: loadData/data_reader.py: change datasets path.

python main.py -pc -pdi -sr

Installation

This project is implemented with Pytorch and has been tested on version

  • Pytorch 1.7,
  • numpy 1.21.4,
  • matplotlib 3.3.3
  • scikit-learn 0.23.2.

Citation

Please kindly cite the papers Weighted Feature Fusion of Convolutional Neural Network and Graph Attention Network for Hyperspectral Image Classification if this code is useful and helpful for your research.

@ARTICLE{9693311,
  author={Dong, Yanni and Liu, Quanwei and Du, Bo and Zhang, Liangpei},
  journal={IEEE Transactions on Image Processing}, 
  title={Weighted Feature Fusion of Convolutional Neural Network and Graph Attention Network for Hyperspectral Image Classification}, 
  year={2022},
  volume={31},
  number={},
  pages={1559-1572},
  doi={10.1109/TIP.2022.3144017}}

wfcg's People

Contributors

quanweiliu avatar

Stargazers

 avatar  avatar Angus avatar  avatar Allen Bluce avatar  avatar  avatar  avatar  avatar jg_zhao avatar José Dixon avatar Bard avatar  avatar XuanLiu avatar  avatar  avatar  avatar  avatar  avatar Luo Li avatar Zhou Min avatar  avatar  avatar Foga avatar Zhou Hao avatar Coconut avatar  avatar  avatar  avatar Linexus avatar  avatar Flipped avatar binghao Ye avatar  avatar  avatar  avatar  avatar Zhe Meng avatar  avatar  avatar  avatar CongYang avatar  avatar  avatar  avatar Perseverance avatar  avatar  avatar gorgeousyyy avatar Mingsong Li avatar

Watchers

 avatar

wfcg's Issues

模型训练占用显存较大

你好,我使用16G的RTX3080显卡设备试着跑了一下PU数据集和HongHu数据集,发现显存不够。究其原因,一方面是因为将HSI全影像输入,没有做batch_size处理,占用显存较大。另一方面是因为GAT和PAM处计算自注意力机制会占用较大的显存。请问你使用的是多大显存的RTX3090卡呢?

Visualization of classification results

Hello author, after carefully reading your paper, I have researched the relevant code for your paper. However, I have found that no matter how I make modifications, I still cannot achieve the visualization of classification results in your paper. I have made many modifications, but in the end, I still generate label images. So, I would like to ask how you achieve the visualization of classification results

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.