Shortcuts

mmcv.cnn

Module

ContextBlock

ContextBlock module in GCNet.

Conv2d

Conv3d

ConvAWS2d

AWS (Adaptive Weight Standardization)

ConvModule

A conv block that bundles conv/norm/activation layers.

ConvTranspose2d

ConvTranspose3d

ConvWS2d

DepthwiseSeparableConvModule

Depthwise separable convolution module.

GeneralizedAttention

GeneralizedAttention module.

HSigmoid

Hard Sigmoid Module.

HSwish

Hard Swish Module.

Linear

MaxPool2d

MaxPool3d

NonLocal1d

1D Non-local module.

NonLocal2d

2D Non-local module.

NonLocal3d

3D Non-local module.

Scale

A learnable scale parameter.

Swish

Swish Module.

Conv2dRFSearchOp

Enable Conv2d with receptive field searching ability.

Build Function

build_activation_layer

Build activation layer.

build_conv_layer

Build convolution layer.

build_norm_layer

Build normalization layer.

build_padding_layer

Build padding layer.

build_plugin_layer

Build plugin layer.

build_upsample_layer

Build upsample layer.

Miscellaneous

fuse_conv_bn

Recursively fuse conv and bn in a module.

conv_ws_2d

is_norm

Check if a layer is a normalization layer.

make_res_layer

make_vgg_layer

get_model_complexity_info

Get complexity information of a model.