Skip to content
/ UNET Public

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

License

Notifications You must be signed in to change notification settings

mikami520/UNET

Repository files navigation

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

Star History Chart

About

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

Topics

Resources

License

Stars

Watchers

Forks

Languages