Skip to content

Commit

Permalink
Update torchvision ops in doc (#2341)
Browse files Browse the repository at this point in the history
* Add Deformable Convolution in Doc

* Add PSRoIPool and PSRoIAlign in Doc
  • Loading branch information
zhiqwang authored Jun 23, 2020
1 parent 42aa9b2 commit b6f55ed
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions docs/source/ops.rst
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,15 @@ torchvision.ops

.. autofunction:: nms
.. autofunction:: roi_align
.. autofunction:: ps_roi_align
.. autofunction:: roi_pool
.. autofunction:: ps_roi_pool
.. autofunction:: deform_conv2d

.. autoclass:: RoIAlign
.. autoclass:: PSRoIAlign
.. autoclass:: RoIPool
.. autoclass:: PSRoIPool
.. autoclass:: DeformConv2d
.. autoclass:: MultiScaleRoIAlign
.. autoclass:: FeaturePyramidNetwork

0 comments on commit b6f55ed

Please sign in to comment.