delta.model.unet_rois

delta.model.unet_rois(input_size: Tuple[int, int, int] = (512, 512, 1), levels: int = 5) → Model

ROIs segmentation U-Net.

Parameters

input_sizetuple of 3 ints, optional

Dimensions of the input tensor, without batch size. The default is (512,512,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

ROIs ID U-Net (compiled).