o0t1ng0o / srd-sai

The source code of 'Super-Resolution Enhanced Medical Image Diagnosis with Sample Affinity Interaction' (TMI 2021).

SRD-SAI

This repository is an official PyTorch implementation of the paper "Super-Resolution Enhanced Medical Image Diagnosis with Sample Affinity Interaction" [paper] from TMI 2021.

Dependencies

• Python 3.6
• PyTorch >= 1.3.0
• numpy

Quickstart

• Train the SRD-SAI framework:

python ./train_srd.py --job_type S --batch_size 8 --optim Adam --num_workers 4 --lr 5e-4 --lr_decay_interval 30 --lr_decay_gamma 0.5 --weight_decay 1e-5 --epochs 350 --dual_ratio 100 --regular_ratio 100 --rank_ratio 1 --ensemble_ce_ratio 2.0 --super_ce_ratio 1.0 --low_ce_ratio 1.0 --aux_ce_ratio 1.0 --resolution 128 --cross_str cross0 --lr_factor_srnet 0.01

Cite

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

@article{chen2021super,
  title={Super-Resolution Enhanced Medical Image Diagnosis with Sample Affinity Interaction},
  author={Chen, Zhen and Guo, Xiaoqing and Woo, Peter YM and Yuan, Yixuan},
  journal={IEEE Transactions on Medical Imaging},
  year={2021},
  publisher={IEEE}
}