diff --git a/torchvision/ops/__init__.py b/torchvision/ops/__init__.py index a047f26c321..0acd5fe990f 100644 --- a/torchvision/ops/__init__.py +++ b/torchvision/ops/__init__.py @@ -24,6 +24,7 @@ __all__ = [ + "masks_to_boxes", "deform_conv2d", "DeformConv2d", "nms",