DeformConv2dPack

class mmcv.ops.DeformConv2dPack(*args, **kwargs)

A Deformable Conv Encapsulation that acts as normal Conv layers.

The offset tensor is like [y0, x0, y1, x1, y2, x2, …, y8, x8]. The spatial arrangement is like:

(x0, y0) (x1, y1) (x2, y2)
(x3, y3) (x4, y4) (x5, y5)
(x6, y6) (x7, y7) (x8, y8)

参数

• in_channels (int) – Same as nn.Conv2d.

• out_channels (int) – Same as nn.Conv2d.

• kernel_size (int or tuple[int]) – Same as nn.Conv2d.

• stride (int or tuple[int]) – Same as nn.Conv2d.

• padding (int or tuple[int]) – Same as nn.Conv2d.

• dilation (int or tuple[int]) – Same as nn.Conv2d.

• groups (int) – Same as nn.Conv2d.

• bias (bool or str) – If specified as auto, it will be decided by the norm_cfg. Bias will be set as True if norm_cfg is None, otherwise False.

forward(x: torch.Tensor) → torch.Tensor

Deformable Convolutional forward function.

参数

• x (Tensor) – Input feature, shape (B, C_in, H_in, W_in)

• offset (Tensor) –

  Offset for deformable convolution, shape (B, deform_groups*kernel_size[0]*kernel_size[1]*2, H_out, W_out), H_out, W_out are equal to the output’s.

  An offset is like [y0, x0, y1, x1, y2, x2, …, y8, x8]. The spatial arrangement is like:

  (x0, y0) (x1, y1) (x2, y2)
  (x3, y3) (x4, y4) (x5, y5)
  (x6, y6) (x7, y7) (x8, y8)
  

返回

Output of the layer.

返回类型

Tensor