vis4d.op.layer.util

Utility functions for layer ops.

Functions

`build_activation_layer`(activation[, inplace])	Build activation layer.
`build_conv_layer`(in_planes, out_planes[, ...])	Build a convolution layer.
`build_norm_layer`(norm, out_channels[, ...])	Build normalization layer.

build_conv_layer(in_planes, out_planes, kernel_size=3, stride=1, padding=0, dilation=1, groups=1, bias=False, norm=None, activation=None, use_dcn=False)[source]

Build a convolution layer.

Return type:: Module

build_activation_layer(activation, inplace=False)[source]

Build activation layer.

Parameters:

activation (str) – Activation layer type.
inplace (bool, optional) – If to set inplace. Defaults to False. It will be ignored if the activation layer is not inplace.

Return type:

Module

build_norm_layer(norm, out_channels, num_groups=None)[source]

Build normalization layer.

Parameters:

norm (str) – Normalization layer type.
out_channels (int) – Number of output channels.
num_groups (int | None, optional) – Number of groups for GroupNorm. Defaults to None.

Return type:

Module