Adds docs and Fixes Imports for ops and io (pytorch#2654)

* Adds docs and fixes improts for ops and io * reverts io and init for images
Quansight · Dec 4, 2020 · a7d9494 · a7d9494
1 parent cf0b9d4
commit a7d9494
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/docs/source/ops.rst b/docs/source/ops.rst
@@ -10,6 +10,11 @@ torchvision.ops
 
 
 .. autofunction:: nms
+.. autofunction:: batched_nms
+.. autofunction:: remove_small_boxes
+.. autofunction:: clip_boxes_to_image
+.. autofunction:: box_area
+.. autofunction:: box_iou
 .. autofunction:: roi_align
 .. autofunction:: ps_roi_align
 .. autofunction:: roi_pool

diff --git a/torchvision/io/__init__.py b/torchvision/io/__init__.py
@@ -15,7 +15,6 @@
     write_video,
 )
 
-
 __all__ = [
     "write_video",
     "read_video",

diff --git a/torchvision/ops/__init__.py b/torchvision/ops/__init__.py
@@ -1,4 +1,4 @@
-from .boxes import nms, box_iou
+from .boxes import nms, batched_nms, remove_small_boxes, clip_boxes_to_image, box_area, box_iou
 from .new_empty_tensor import _new_empty_tensor
 from .deform_conv import deform_conv2d, DeformConv2d
 from .roi_align import roi_align, RoIAlign
@@ -14,7 +14,8 @@
 
 
 __all__ = [
-    'deform_conv2d', 'DeformConv2d', 'nms', 'roi_align', 'RoIAlign', 'roi_pool',
+    'deform_conv2d', 'DeformConv2d', 'nms', 'batched_nms', 'remove_small_boxes',
+    'clip_boxes_to_image', 'box_area', 'box_iou', 'roi_align', 'RoIAlign', 'roi_pool',
     'RoIPool', '_new_empty_tensor', 'ps_roi_align', 'PSRoIAlign', 'ps_roi_pool',
     'PSRoIPool', 'MultiScaleRoIAlign', 'FeaturePyramidNetwork'
 ]