pytorch safetensors ProbUNet medical probabilistic unet 2D PULASki vessel segmentation 7T MRA-ToF MRA TOF MRI 7T Conditional VAE distribution distance image-segmentation custom_code arxiv:2312.15686 license:apache-2.0 region:us
Small (<1GB)