delta.model.unet_seg

delta.model.unet_seg(pretrained_weights: Optional[str] = None, input_size: Tuple[int, int, int] = (256, 32, 1), levels: int = 5) → Model

Cell segmentation U-Net definition function.

Parameters

pretrained_weightshdf5 file, optional

Model will load weights from hdf5 and start training. The default is None

input_sizetuple of 3 ints, optional

Dimensions of the input tensor, without batch size. The default is (256,32,1).

levelsint, optional

Number of levels of the U-Net, ie number of successive contraction then expansion blocks are combined together. The default is 5.

Returns

modelModel

Segmentation U-Net (compiled).