/hsid-cnn-pytorch

PyTorch version of HSID-CNN. Original paper: Q. Yuan, Q. Zhang, J. Li, H. Shen, and L. Zhang, "Hyperspectral Image Denoising Employing a Spatial-Spectral Deep Residual Convolutional Neural Network", IEEE TGRS, 2019.

Primary LanguagePython

PyTorch version of HSID-CNN

Official version(Caffe + matlab): https://github.com/qzhang95/HSID-CNN

Environment

  • Python 3, PyTorch 1.7
  • numpy, scipy, tqdm
  • Ubuntu 18, cuda 10.2

Introduction

  • Network architecture in network.py.
  • Other python files are written for training and evaluation.