by Quande Liu, Lequan Yu, Luyang Luo, Qi Dou, Pheng-Ann Heng.

This repository is for our TMI 2020 paper 'Semi-supervised Medical Image Classification with Relation-driven Self-ensembling Model'.

This repository is based on PyTorch 1.4.0.

1. Clone the repository:

   git clone https://github.com/liuquande/SRC-MT.git
   cd code

2. Organize the dataset

3. Train the model:

   python train_SRC_MT.py --gpu 0,1,2

If this repository is useful for your research, please cite:

@article{liu2020semi,
  title={Semi-supervised medical image classification with relation-driven self-ensembling model},
  author={Liu, Quande and Yu, Lequan and Luo, Luyang and Dou, Qi and Heng, Pheng Ann},
  journal={IEEE Transactions on Medical Imaging},
  year={2020},
  publisher={IEEE}
}

Please contact '[email protected]'