Skip to content

Commit b6f55ed

Browse files
authored
Update torchvision ops in doc (#2341)
* Add Deformable Convolution in Doc * Add PSRoIPool and PSRoIAlign in Doc
1 parent 42aa9b2 commit b6f55ed

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

docs/source/ops.rst

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,15 @@ torchvision.ops
1111

1212
.. autofunction:: nms
1313
.. autofunction:: roi_align
14+
.. autofunction:: ps_roi_align
1415
.. autofunction:: roi_pool
16+
.. autofunction:: ps_roi_pool
17+
.. autofunction:: deform_conv2d
1518

1619
.. autoclass:: RoIAlign
20+
.. autoclass:: PSRoIAlign
1721
.. autoclass:: RoIPool
22+
.. autoclass:: PSRoIPool
23+
.. autoclass:: DeformConv2d
24+
.. autoclass:: MultiScaleRoIAlign
25+
.. autoclass:: FeaturePyramidNetwork

0 commit comments

Comments
 (0)