From 9445e36fc4bf30bc5fec394aa36ec6e643095b7d Mon Sep 17 00:00:00 2001 From: Joao Gomes Date: Thu, 28 Oct 2021 16:06:39 +0100 Subject: [PATCH] Adding masks_to_boxes to __all__ in ops --- torchvision/ops/__init__.py | 1 + 1 file changed, 1 insertion(+) 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",