napari_cellseg3d.code_models.models#

Contains model code and wrappers for the models, as classes.

Modules

napari_cellseg3d.code_models.models.TEMPLATE_model

This is a template for a model class.

napari_cellseg3d.code_models.models.model_SegResNet

SegResNet wrapper for napari_cellseg3d.

napari_cellseg3d.code_models.models.model_SwinUNetR

SwinUNetR wrapper for napari_cellseg3d.

napari_cellseg3d.code_models.models.model_TRAILMAP

Legacy version of adapted TRAILMAP model, not used in the current version of the plugin.

napari_cellseg3d.code_models.models.model_TRAILMAP_MS

TRAILMAP model, reimplemented in PyTorch.

napari_cellseg3d.code_models.models.model_VNet

VNet wrapper for napari_cellseg3d.

napari_cellseg3d.code_models.models.model_WNet

Wrapper for the W-Net model, with the decoder weights removed.

napari_cellseg3d.code_models.models.model_test

Model for testing purposes.

napari_cellseg3d.code_models.models.pretrained

Hosts the downloaded pretrained model weights.

napari_cellseg3d.code_models.models.unet

Building block of a UNet model.

napari_cellseg3d.code_models.models.wnet

Building blocks for WNet model.