mikami520/UNET

A Simple Tutorial for 2D and 3D U-Net Training and Test Pipeline in Pytorch

A-Simple-2D&3D-UNet-Pytorch-Tutorial

The results trained and tested on Liver dataset from Medical Segmentation Decathlon. Checkpoints, loss plots, inference results etc are stored in the 3D UNet Liver and 2D UNet Liver

Metrics

Multiple evaluation metrics are added to this tutorial:

• Mean Dice Score
• Mean Surface Distance
• Mean Hausdorff Distance
• Mean Surface Dice Score
• Mean Chamfer Distance

Star History