Home

Awesome

Dual-Task Mutual Learning for Semi-Supervised Medical Image Segmentation

Introduction

This is the repository of 'Dual-Task Mutual Learning for Semi-Supervised Medical Image Segmentation' (PRCV 2021)

Usage

  1. Clone the repo
git clone https://github.com/YichiZhang98/DTML
cd DTML
  1. Put the data in data/2018LA_Seg_Training Set.

  2. Train the model

cd code
python train_la_dtml.py
  1. Test the model
python test_LA.py

Acknowledgement

@inproceedings{zhang2021dual,
  title={Dual-Task Mutual Learning for Semi-Supervised Medical Image Segmentation},
  author={Zhang, Yichi and Zhang, Jicong},
  booktitle={Pattern Recognition and Computer Vision},
  pages={548--559},
  year={2021}
}