Adding masks_to_boxes to __all__ in ops (#4779)

Adding masks_to_boxes to all in ops (#4779)
e0a5fede · Joao Gomes · GitHub · 7d956458 · e0a5fede
Unverified Commit e0a5fede authored Oct 29, 2021 by Joao Gomes Committed by GitHub Oct 29, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

torchvision/ops/__init__.py torchvision/ops/__init__.py +1 -0

No files found.
--- a/torchvision/ops/__init__.py
+++ b/torchvision/ops/__init__.py
@@ -24,6 +24,7 @@ _register_custom_op()


 __all__ = [
+    "masks_to_boxes",
    "deform_conv2d",
    "DeformConv2d",
    "nms",