/SM-CNN

Primary LanguagePython

SM-CNN: Hyperspectral Image Denoising via Self-Modulating Convolutional Neural Networks

Paper

Installation

  • Install dependencies:
pip install -r requirements.txt

Train and Test

  • Put your data into the 'datasets' directory
  • Creat train and val patches:
python get_patch_wdc.py
  • Creat test data:
python get_patch_wdc_test.py
  • Open the model directory in terminal
  • Set python environment
  • Train example:
python .\train.py --config-file .\gauss_blind_config.yaml
  • Test example:
python .\predict.py --config-file .\gauss_blind_config.yaml