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. The model performs unsupervised segmentation of 3D images.

Classes

Block(in_channels, out_channels[, dropout])	Basic block of the U-Net architecture.
InBlock(in_channels, out_channels[, dropout])	Input block of the U-Net architecture.
OutBlock(in_channels, out_channels[, dropout])	Output block of the U-Net architecture.
UNet(in_channels, out_channels[, channels, ...])	Half of the W-Net model, based on the U-Net architecture.
WNet([in_channels, out_channels, ...])	Implementation of a 3D W-Net model, based on the 2D version from https://arxiv.org/abs/1711.08506.
WNet_encoder([in_channels, out_channels, ...])	WNet3D with encoder only.