ops.rst 948 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
13
.. autofunction:: batched_nms
.. autofunction:: box_area
14
.. autofunction:: box_convert
15
.. autofunction:: box_iou
16
17
.. autofunction:: clip_boxes_to_image
.. autofunction:: deform_conv2d
Aditya Oke's avatar
Aditya Oke committed
18
.. autofunction:: generalized_box_iou
19
20
.. autofunction:: masks_to_boxes
.. autofunction:: nms
21
22
.. autofunction:: ps_roi_align
.. autofunction:: ps_roi_pool
23
24
25
.. autofunction:: remove_small_boxes
.. autofunction:: roi_align
.. autofunction:: roi_pool
Aditya Oke's avatar
Aditya Oke committed
26
.. autofunction:: sigmoid_focal_loss
27
.. autofunction:: stochastic_depth
28
29

.. autoclass:: RoIAlign
30
.. autoclass:: PSRoIAlign
31
.. autoclass:: RoIPool
32
33
34
35
.. autoclass:: PSRoIPool
.. autoclass:: DeformConv2d
.. autoclass:: MultiScaleRoIAlign
.. autoclass:: FeaturePyramidNetwork
36
.. autoclass:: StochasticDepth