tfm.vision.backbones.DilatedResNet

Creates a ResNet model with Deeplabv3 modifications.

tfm.vision.backbones.DilatedResNet(
    model_id: int,
    output_stride: int,
    input_specs: tf.keras.layers.InputSpec = layers.InputSpec(shape=[None, None, None, 3]),
    stem_type: str = 'v0',
    resnetd_shortcut: bool = False,
    replace_stem_max_pool: bool = False,
    se_ratio: Optional[float] = None,
    init_stochastic_depth_rate: float = 0.0,
    multigrid: Optional[Tuple[int]] = None,
    last_stage_repeats: int = 1,
    activation: str = 'relu',
    use_sync_bn: bool = False,
    norm_momentum: float = 0.99,
    norm_epsilon: float = 0.001,
    kernel_initializer: str = 'VarianceScaling',
    kernel_regularizer: Optional[tf.keras.regularizers.Regularizer] = None,
    bias_regularizer: Optional[tf.keras.regularizers.Regularizer] = None,
    **kwargs
)

This backbone is suitable for semantic segmentation. This implements Liang-Chieh Chen, George Papandreou, Florian Schroff, Hartwig Adam. Rethinking Atrous Convolution for Semantic Image Segmentation. (https://arxiv.org/pdf/1706.05587)

Args
model_id	An int specifies depth of ResNet backbone model.
output_stride	An int of output stride, ratio of input to output resolution.
input_specs	A tf.keras.layers.InputSpec of the input tensor.
stem_type	A str of stem type. Can be v0 or v1. v1 replaces 7x7 conv by 3 3x3 convs.
resnetd_shortcut	A bool of whether to use ResNet-D shortcut in downsampling blocks.
replace_stem_max_pool	A bool of whether to replace the max pool in stem with a stride-2 conv,
se_ratio	A float or None. Ratio of the Squeeze-and-Excitation layer.
init_stochastic_depth_rate	A float of initial stochastic depth rate.
multigrid	A tuple of the same length as the number of blocks in the last resnet stage.
last_stage_repeats	An int that specifies how many times last stage is repeated.
activation	A str name of the activation function.
use_sync_bn	If True, use synchronized batch normalization.
norm_momentum	A float of normalization momentum for the moving average.
norm_epsilon	A float added to variance to avoid dividing by zero.
kernel_initializer	A str for kernel initializer of convolutional layers.
kernel_regularizer	A tf.keras.regularizers.Regularizer object for Conv2D. Default to None.
bias_regularizer	A tf.keras.regularizers.Regularizer object for Conv2D. Default to None.
**kwargs	Additional keyword arguments to be passed.

Attributes
output_specs	A dict of {level: TensorShape} pairs for the model output.

Methods

call

call(
    inputs, training=None, mask=None
)

Calls the model on new inputs and returns the outputs as tensors.

In this case call() just reapplies all ops in the graph to the new inputs (e.g. build a new computational graph from the provided inputs).

Note: This method should not be called directly. It is only meant to be overridden when subclassing tf.keras.Model. To call a model on an input, always use the __call__() method, i.e. model(inputs), which relies on the underlying call() method.

Args
inputs	Input tensor, or dict/list/tuple of input tensors.
training	Boolean or boolean scalar tensor, indicating whether to run the Network in training mode or inference mode.
mask	A mask or list of masks. A mask can be either a boolean tensor or None (no mask). For more details, check the guide here.

Returns
A tensor if there is a single output, or a list of tensors if there are more than one outputs.