ops.rst 782 Bytes
Newer Older
1
2
3
4
5
6
7
8
torchvision.ops
===============

.. currentmodule:: torchvision.ops

:mod:`torchvision.ops` implements operators that are specific for Computer Vision.

.. note::
9
  All operators have native support for TorchScript.
10
11
12


.. autofunction:: nms
13
14
15
16
17
.. autofunction:: batched_nms
.. autofunction:: remove_small_boxes
.. autofunction:: clip_boxes_to_image
.. autofunction:: box_area
.. autofunction:: box_iou
Aditya Oke's avatar
Aditya Oke committed
18
.. autofunction:: generalized_box_iou
19
.. autofunction:: roi_align
20
.. autofunction:: ps_roi_align
21
.. autofunction:: roi_pool
22
23
.. autofunction:: ps_roi_pool
.. autofunction:: deform_conv2d
24
25

.. autoclass:: RoIAlign
26
.. autoclass:: PSRoIAlign
27
.. autoclass:: RoIPool
28
29
30
31
.. autoclass:: PSRoIPool
.. autoclass:: DeformConv2d
.. autoclass:: MultiScaleRoIAlign
.. autoclass:: FeaturePyramidNetwork