torch_em.model
Neural network architectures for classifcation, segmentation or image-to-image translation tasks.
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