cellmap_segmentation_challenge.models# Modules model_load resnet unet_model_2D Parts of the U-Net model unet_model_3D Parts of the U-Net model vitnet