NonLocal3d

class mmcv.cnn.NonLocal3d(in_channels: int, sub_sample: bool = False, conv_cfg: Dict = {'type': 'Conv3d'}, **kwargs)

3D Non-local module.

Parameters

• in_channels (int) – Same as NonLocalND.

• sub_sample (bool) – Whether to apply max pooling after pairwise function (Note that the sub_sample is applied on spatial only). Default: False.

• conv_cfg (None | dict) – Same as NonLocalND. Default: dict(type=’Conv3d’).