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
- Clone the repo
git clone https://github.com/YichiZhang98/DTML
cd DTML
-
Put the data in data/2018LA_Seg_Training Set.
-
Train the model
cd code
python train_la_dtml.py
- Test the model
python test_LA.py
Acknowledgement
- This code is adapted from UA-MT, SegWithDistMap and DTC. We thank all the authors for their contribution.
- If our work is useful for your research, please consider citing:
@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}
}