delta.model.unet_seg#
- delta.model.unet_seg(pretrained_weights=None, input_size=(256, 32, 1), levels=5, filters=None)[source]#
Cell segmentation U-Net definition function.
- Parameters
- pretrained_weightsmodel file, optional
Model will load weights from file 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. Ignored if filters is specified. The default is 5.
- filters[int], optional
Number of convolutional kernels at each level. The default is starting with 64 and multiplying by 2 at each level.
- Returns
- modelModel
Segmentation U-Net (compiled).