napari_cellseg3d.code_models.models.wnet#

Building blocks for WNet model.

Modules

napari_cellseg3d.code_models.models.wnet.model

Implementation of a 3D W-Net model, based on the 2D version from https://arxiv.org/abs/1711.08506.

napari_cellseg3d.code_models.models.wnet.soft_Ncuts

Implementation of a 3D Soft N-Cuts loss based on https://arxiv.org/abs/1711.08506 and https://ieeexplore.ieee.org/document/868688.