/CAML

repository for CAML: Correlation-aware mutual learning for semi-supervised medical image segmentation

Primary LanguagePython

Correlation-Aware Mutual Learning for Semi-supervised Medical Image Segmentation

Introduction

This repository is for our paper: 'Correlation-Aware Mutual Learning for Semi-supervised Medical Image Segmentation'.

Requirements

This repository is based on PyTorch 1.8.1, CUDA 10.2 and Python 3.6.9; All experiments in our paper were conducted on a single NVIDIA TITAN RTX GPU.

Usage

  1. Clone the repo.;
https://github.com/Herschel555/CAML.git

  1. Put the data in './CAML/data';

  2. Train the model;

cd CAML
# e.g., for 5% labels on LA
python code/train_caml.py --labelnum 4 --gpu 0 --seed 1337
  1. Test the model;
cd CAML
# e.g., for 5% labels on LA
python code/test_3d.py --labelnum 4 --gpu 0 --seed 1337

Acknowledgements:

Our code is origin from UAMT, SSL4MIS and MC-Net. Thanks for these authors for their valuable works.