torch_em.model

Neural network architectures for classifcation, segmentation or image-to-image translation tasks.

View Source

1"""Neural network architectures for classifcation, segmentation or image-to-image translation tasks.
2"""
3
4from .unet import AnisotropicUNet, UNet2d, UNet3d
5from .probabilistic_unet import ProbabilisticUNet
6from .unetr import UNETR
7from .vit import get_vision_transformer
8from .vim import get_vimunet_model