- Ubuntu 20.04
- PyTorch 1.11.0
- medpy, tqdm, h5py, etc.
# after install dependcies
git clone [email protected]:Axi404/PMT.git
cd PMT/code
python train_PMT.py
python test_LA.pyFollowing the setting of MCF.
We build the project based on UA-MT, SASSNet, DTC, MCF. Thanks for their contribution.
If you find this project useful, please consider citing:
@article{gao2024pmtprogressivemeanteacher,
title={PMT: Progressive Mean Teacher via Exploring Temporal Consistency for Semi-Supervised Medical Image Segmentation},
author={Ning Gao and Sanping Zhou and Le Wang and Nanning Zheng},
journal={arXiv preprint arXiv:2409.05122},
year={2024}
}