delta.model.unet_rois#
- delta.model.unet_rois(input_size=(512, 512, 1), levels=5, filters=None)[source]#
Segmentation U-Net for ROIs.
- 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. 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
ROIs ID U-Net (compiled).