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