jqyang22 / hsic-fm

[TNNLS 2023] Overcoming the Barrier of Incompleteness: A Hyperspectral Image Classification Full Model

Overcoming the Barrier of Incompleteness: A Hyperspectral Image Classification Full Model

PyTorch implementation of HSIC-FM for hyperspectral image classification.

Installation

Install Pytorch 1.9.0 with Python 3.8.

Usage

Please first change the path in the code, and then run:
python ./main.py --lstm

Paper

Overcoming the Barrier of Incompleteness: A Hyperspectral Image Classification Full Model
Please kindly cite our paper if you find it's helpful in your work. Thank you!

hsic-fm's Issues

CUDA_LAUNCH_BLOCKING=1

Hello,my device is also a 3090, but I made a change in the code here (GPU_ID = '0'), I encountered the following error while trying to reproduce your code,:
File "./HSIC-FM-main-2023-weishixian-envs\main.py", line 217, in
loss_trn, acc_trn, tsne_trn, tsne_trn_y = b.train(i, loss_trn, acc_trn, tsne_trn, tsne_trn_y, net, optimizer, scheduler, trn_loader, criterion)
File "e:\hjj\HSIC-FM-main-2023-weishixian-envs\network.py", line 37, in train
epochavg_loss += loss.item()
RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1

I have made many attempts but have been unable to solve this problem. Do you know how to resolve it?

local variable 'net' referenced before assignment

File "/content/gdrive/MyDrive/research/main.py", line 186, in
net = FullModel(
File "/content/gdrive/MyDrive/research/HSIC_FM.py", line 180, in init
self.recurrent = net(input_size, args.d, args.depth)
UnboundLocalError: local variable 'net' referenced before assignment