napari_cellseg3d.code_models.models#
Contains model code and wrappers for the models, as classes.
Modules
This is a template for a model class. |
|
SegResNet wrapper for napari_cellseg3d. |
|
SwinUNetR wrapper for napari_cellseg3d. |
|
Legacy version of adapted TRAILMAP model, not used in the current version of the plugin. |
|
TRAILMAP model, reimplemented in PyTorch. |
|
VNet wrapper for napari_cellseg3d. |
|
Wrapper for the W-Net model, with the decoder weights removed. |
|
Model for testing purposes. |
|
Hosts the downloaded pretrained model weights. |
|
Building block of a UNet model. |
|
Building blocks for WNet model. |