Skip to content

[PRCV 2021] The official code for "Dual-Task Mutual Learning for Semi-Supervised Medical Image Segmentation"

Notifications You must be signed in to change notification settings

YichiZhang98/DTML

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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

  • 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}
}

About

[PRCV 2021] The official code for "Dual-Task Mutual Learning for Semi-Supervised Medical Image Segmentation"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages