Hyper W-Net¶

class src.pipelines.tensorflow_v2.models.hyper_wnet.ConvBridgeBlock(*args, **kwargs)[source]¶

A Convolutional Bridge Block to be used in a W-Net. This is similar to the class defined in the wnet.py script, but has the addition of the kernel_size in the instantiator to allow the option to include this parameter in the search

call(x)[source]¶

Applies a ConvBridgeBlock to the an input

Parameters: x (Tensor) – the input to apply the ConvBridgeBlock to
Return type: Tensor
Returns: the output of the ConvBridgeBlock

class src.pipelines.tensorflow_v2.models.hyper_wnet.HyperWnet(input_shape, output_channels)[source]¶

A Hyper W-Net model which can be tuned by kerastuner

build(hp)[source]¶

A function which creates a kerastuner hyper model with a defined search space. This function signature matches the requirements of the hyperparameter tuning algorithms in kerastuner.

Parameters: hp (HyperParameters) – a HyperParameters instance
Return type: HyperModel
Returns: a HyperWnet instance