Python PyTorch deform_conv2d用法及代碼示例

本文簡要介紹python語言中 torchvision.ops.deform_conv2d 的用法。

用法:

torchvision.ops.deform_conv2d(input: torch.Tensor, offset: torch.Tensor, weight: torch.Tensor, bias: Optional[torch.Tensor] = None, stride: Tuple[int, int] =(1, 1), padding: Tuple[int, int] =(0, 0), dilation: Tuple[int, int] =(1, 1), mask: Optional[torch.Tensor] = None) → torch.Tensor

參數：

• input(Tensor[batch_size,in_channels,in_height,in_width]) - 輸入張量

• offset(Tensor[batch_size,2 * offset_groups * kernel_height * kernel_width,out_height,out_width]) - 應用於卷積核中每個位置的偏移量。

• weight(Tensor[out_channels,in_channels //組,kernel_height,kernel_width]) - 卷積權重，分為大小組 (in_channels //組)

• bias(Tensor[out_channels]) -可選的形狀偏差(out_channels，)。默認值：無

• stride(int或者元組[int,int]) - 卷積中心之間的距離。默認值：1

• padding(int或者元組[int,int]) - 每個圖像周圍零填充的高度/寬度。默認值：0

• dilation(int或者元組[int,int]) - 內核元素之間的間距。默認值：1

• mask(Tensor[batch_size,offset_groups * kernel_height * kernel_width,out_height,out_width]) - 應用於卷積核中每個位置的掩碼。默認值：無

返回：

卷積的結果

返回類型：

張量[batch_sz, out_channels, out_h, out_w]

如果 mask 不是 None，則執行 Deformable ConvNets v2: More Deformable, Better Results 中說明的可變形卷積 v2；如果 mask 是 None，則執行 Deformable Convolutional Networks 中說明的可變形卷積。

例子：：

>>> input = torch.rand(4, 3, 10, 10)
>>> kh, kw = 3, 3
>>> weight = torch.rand(5, 3, kh, kw)
>>> # offset and mask should have the same spatial size as the output
>>> # of the convolution. In this case, for an input of 10, stride of 1
>>> # and kernel size of 3, without padding, the output size is 8
>>> offset = torch.rand(4, 2 * kh * kw, 8, 8)
>>> mask = torch.rand(4, kh * kw, 8, 8)
>>> out = deform_conv2d(input, offset, weight, mask=mask)
>>> print(out.shape)
>>> # returns
>>>  torch.Size([4, 5, 8, 8])