mmcv.cnn.build_conv_layer

mmcv.cnn.build_conv_layer(cfg: Optional[Dict], *args, **kwargs) → torch.nn.modules.module.Module

Build convolution layer.

参数

• cfg (None or dict) – The conv layer config, which should contain: - type (str): Layer type. - layer args: Args needed to instantiate an conv layer.

• args (argument list) – Arguments passed to the __init__ method of the corresponding conv layer.

• kwargs (keyword arguments) – Keyword arguments passed to the __init__ method of the corresponding conv layer.

返回

Created conv layer.

返回类型

nn.Module