delta.model.unet_track

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

Tracking 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,4).

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

Tracking U-Net (compiled).