delta.model.unet_track

delta.model.unet_track(pretrained_weights: Optional[str] = None, input_size: Tuple[int, int, int] = (256, 32, 4)) → keras.engine.training.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).

Returns

modelModel

Tracking U-Net (compiled).