UNet2D model for PyTorch
Application
Computer Vision, Segmentation
Framework
PyTorch
GitHub Link
Pretrained Models
Checkpoints
SynapseAI 1.11.0
https://vault.habana.ai/artifactory/pretrained-models/checkpoints/1.11.0/PT/Unet2d/Unet2d-PyTorch-2.0.1-1.11.0-587-8n-checkpoint.tar.gz
Datasets
Medical Segmentation Decathlon – BraTS
License
CC BY 4.0