mmcv.cnn.build_upsample_layer

mmcv.cnn.build_upsample_layer(cfg: Dict, *args, **kwargs) → torch.nn.modules.module.Module

Build upsample layer.

参数

• cfg (dict) –

  The upsample layer config, which should contain:

  • type (str): Layer type.

  • scale_factor (int): Upsample ratio, which is not applicable to deconv.

  • layer args: Args needed to instantiate a upsample 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 upsample layer.

返回类型

nn.Module