Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
ops.rst 839 Bytes
Newer Older
Aditya Oke's avatar
Rewrite gallery example for masks to boxes. (#4484)  
Aditya Oke committed
1
2 
.. _ops:
Nicolas Hug's avatar
Big doc revamp - simplify and improve the subpackage pages (#4783)  
Nicolas Hug committed
3
4 
Operators
=========
Francisco Massa's avatar
Expose docs for io and ops package (#1189)  
Francisco Massa committed
5
6
7
8
9
10 

.. currentmodule:: torchvision.ops

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

.. note::
Zhiqiang Wang's avatar
Update the statement of supporting torchscript ops (#2343)  
Zhiqiang Wang committed
11 
  All operators have native support for TorchScript.
Francisco Massa's avatar
Expose docs for io and ops package (#1189)  
Francisco Massa committed
12
Nicolas Hug's avatar
Big doc revamp - simplify and improve the subpackage pages (#4783)  
Nicolas Hug committed
13
14
15 
.. autosummary::
    :toctree: generated/
    :template: function.rst
Francisco Massa's avatar
Expose docs for io and ops package (#1189)  
Francisco Massa committed
16
Nicolas Hug's avatar
Big doc revamp - simplify and improve the subpackage pages (#4783)  
Nicolas Hug committed
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45 
    batched_nms
    box_area
    box_convert
    box_iou
    clip_boxes_to_image
    deform_conv2d
    generalized_box_iou
    masks_to_boxes
    nms
    ps_roi_align
    ps_roi_pool
    remove_small_boxes
    roi_align
    roi_pool
    sigmoid_focal_loss
    stochastic_depth

.. autosummary::
    :toctree: generated/
    :template: class.rst

    RoIAlign
    PSRoIAlign
    RoIPool
    PSRoIPool
    DeformConv2d
    MultiScaleRoIAlign
    FeaturePyramidNetwork
    StochasticDepth
Joao Gomes's avatar
Expose misc ops at package level (#4812)  
Joao Gomes committed
46
47
48 
    FrozenBatchNorm2d
    ConvNormActivation
    SqueezeExcitation