PLATO

Official pytorch implementation of paper "PLATO: ProbabiListic hierArchical mulTi-head mOdel for Plug-and-play Ambiguous Medical Image Segmentation"

Full code will be publicly available upon acceptence.

Requirements

You can install the required packages by executing the following command

pip install -r requirements.txt

Training

You can train and test a PLATO module with pretrained model by running the codes.