Home

Awesome

Uncertainty-Aware Temporal Self-Learning (UATS)

This repository contains the source code of the paper Uncertainty-Aware Temporal Self-Learning (UATS): Semi-Supervised Learning for Segmentation of the Prostate Zones and Beyond, accepted in Artificial Intelligence in Medicine Journal.

Concept of our method. For details we refer to our paper at .....

Highlights:

Training:

  1. pip install -r requirements.txt
  2. Experiment and dataset-related changes to be made in the file: utility/config.py
  3. The UATS model could be trained: python dataset_specific/prostate/expts/prostate_uats_softmax.py