Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
ops.rst 961 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 
    batched_nms
    box_area
    box_convert
    box_iou
    clip_boxes_to_image
    deform_conv2d
Hu Ye's avatar
New Feature: add DropBlock layer (#5416)  
Hu Ye committed
23
24 
    drop_block2d
    drop_block3d
Nicolas Hug's avatar
Big doc revamp - simplify and improve the subpackage pages (#4783)  
Nicolas Hug committed
25 
    generalized_box_iou
Aditya Oke's avatar
Expose generalized_box_iou to docs (#5302)  
Aditya Oke committed
26 
    generalized_box_iou_loss
Nicolas Hug's avatar
Big doc revamp - simplify and improve the subpackage pages (#4783)  
Nicolas Hug committed
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48 
    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
49 
    FrozenBatchNorm2d
Aditya Oke's avatar
Add Conv2dNormActivation and Conv3dNormActivation Blocks (#5445)  
Aditya Oke committed
50
51 
    Conv2dNormActivation
    Conv3dNormActivation
Joao Gomes's avatar
Expose misc ops at package level (#4812)  
Joao Gomes committed
52 
    SqueezeExcitation
Hu Ye's avatar
New Feature: add DropBlock layer (#5416)  
Hu Ye committed
53
54 
    DropBlock2d
    DropBlock3d