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

* Adds docs and fixes improts for ops and io

* reverts io and init for images

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

torchvision/io/__init__.py

@@ -15,7 +15,6 @@ from .video import (
     write_video,
 )
 
-
 __all__ = [
     "write_video",
     "read_video",

torchvision/ops/__init__.py

-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 @@ _register_custom_op()
 
 
 __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'
 ]