Skip to content

Latest commit

 

History

History
93 lines (78 loc) · 3.71 KB

README.md

File metadata and controls

93 lines (78 loc) · 3.71 KB
Raw

SWD-Net

This repository is an official PyTorch implementation of the paper "Joint Spatial-Wavelet Dual-Stream Network for Super-Resolution" [paper] from MICCAI 2020.

HistoSR

HistoSR dataset is built by random cropping patches from Camelyon16 dataset. By bicubic and nearest downsampling, HistoSR dataset provides a 2× SR from 96×96 pixels to 192×192 pixels with two kinds of degradation. Specifically, the bicubic degradation kernel is the common choice and retains neighboring information, while the nearest one discards the pixels directly. In this way, the nearest version provides a more difficult case to comprehensively evaluate various SR algorithms. Each version of HistoSR dataset contains 30,000 SR pairs in training set and 5,000 SR pairs in test set.

Download

The HistoSR data is stored in LMDB files and can be downloaded from Google Drive. Put the downloaded bicubic and nearest subfolders in a newly-built folder ./HistoSR/.

Dependencies

  • Python 3.6
  • PyTorch >= 1.3.0
  • numpy
  • lmdb

Code

Clone this repository into any place you want.

git clone https://github.com/franciszchen/SWD-Net.git
cd SWD-Net
mkdir save; mkdir HistoSR; mkdir weights

Quickstart

  • Train the SWD-Net with HistoSR bicubic dataset:
python ./train_swdnet.py --theme swdnet-bicubic-default-bsz24 --job_type S --data_degradation bicubic --batch_size 24
  • Train the SWD-Net with HistoSR nearest dataset:
python ./train_swdnet.py --theme swdnet-nearest-default-bsz24 --job_type S --data_degradation nearest --batch_size 24

Benchmark

SWD-Net is implemented and evaluated in RGB-channel. Data augmentation and statistical MeanShift are not employed to optimize SWD-Net.

Method Bicubic degradation Nearest degradation
PSNR SSIM PSNR SSIM
SWD-Net 32.769 0.9510 31.538 0.9397

The weights of SWD-Net to reproduce the records in the paper can be downloaded from Google Drive. Put the downloaded weight files in a newly-built folder ./weights/.

  • For the bicubic degradation:
python ./eval_pth.py --job_type S --data_degradation bicubic --batch_size 24
  • For the nearest degradation:
python ./eval_pth.py --job_type S --data_degradation nearest --batch_size 24

Cite

If you find our work useful in your research or publication, please cite our work:

@inproceedings{chen2020joint,
  title={Joint Spatial-Wavelet Dual-Stream Network for Super-Resolution},
  author={Chen, Zhen and Guo, Xiaoqing and Yang, Chen and Ibragimov, Bulat and Yuan, Yixuan},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={184--193},
  year={2020},
  organization={Springer}
}

Acknowledgements