Unverified Commit afd9d4d8 authored by Haswanth Aekula's avatar Haswanth Aekula Committed by GitHub
Browse files

[DOC] Added paper citation for DeformConv2d (#3000)

* [DOC] Added paper citation for DeformConv2d

* Fixes Pylint error

* Added citation of the previous paper
parent 80f41f8d
...@@ -19,8 +19,13 @@ def deform_conv2d( ...@@ -19,8 +19,13 @@ def deform_conv2d(
dilation: Tuple[int, int] = (1, 1), dilation: Tuple[int, int] = (1, 1),
mask: Optional[Tensor] = None, mask: Optional[Tensor] = None,
) -> Tensor: ) -> Tensor:
""" r"""
Performs Deformable Convolution, described in Deformable Convolutional Networks Performs Deformable Convolution v2, described in
`Deformable ConvNets v2: More Deformable, Better Results
<https://arxiv.org/abs/1811.11168>`__ if :attr:`mask` is not ``None`` and
Performs Deformable Convolution, described in
`Deformable Convolutional Networks
<https://arxiv.org/abs/1703.06211>`__ if :attr:`mask` is ``None``.
Arguments: Arguments:
input (Tensor[batch_size, in_channels, in_height, in_width]): input tensor input (Tensor[batch_size, in_channels, in_height, in_width]): input tensor
...@@ -36,7 +41,7 @@ def deform_conv2d( ...@@ -36,7 +41,7 @@ def deform_conv2d(
dilation (int or Tuple[int, int]): the spacing between kernel elements. Default: 1 dilation (int or Tuple[int, int]): the spacing between kernel elements. Default: 1
mask (Tensor[batch_size, offset_groups * kernel_height * kernel_width, mask (Tensor[batch_size, offset_groups * kernel_height * kernel_width,
out_height, out_width]): masks to be applied for each position in the out_height, out_width]): masks to be applied for each position in the
convolution kernel. convolution kernel. Default: None
Returns: Returns:
output (Tensor[batch_sz, out_channels, out_h, out_w]): result of convolution output (Tensor[batch_sz, out_channels, out_h, out_w]): result of convolution
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment