diff --git a/torchvision/models/detection/generalized_rcnn.py b/torchvision/models/detection/generalized_rcnn.py index c77c892e63e..1acfac29c7a 100644 --- a/torchvision/models/detection/generalized_rcnn.py +++ b/torchvision/models/detection/generalized_rcnn.py @@ -9,6 +9,8 @@ import torch from torch import nn, Tensor +from ...utils import _log_api_usage_once + class GeneralizedRCNN(nn.Module): """ @@ -25,6 +27,7 @@ class GeneralizedRCNN(nn.Module): def __init__(self, backbone, rpn, roi_heads, transform): super(GeneralizedRCNN, self).__init__() + _log_api_usage_once(self) self.transform = transform self.backbone = backbone self.rpn = rpn diff --git a/torchvision/models/detection/retinanet.py b/torchvision/models/detection/retinanet.py index 4b16d7edc7f..98a9ec335d4 100644 --- a/torchvision/models/detection/retinanet.py +++ b/torchvision/models/detection/retinanet.py @@ -11,6 +11,7 @@ from ...ops import boxes as box_ops from ...ops import misc as misc_nn_ops from ...ops.feature_pyramid_network import LastLevelP6P7 +from ...utils import _log_api_usage_once from ..resnet import resnet50 from . import _utils as det_utils from ._utils import overwrite_eps @@ -336,6 +337,7 @@ def __init__( topk_candidates=1000, ): super().__init__() + _log_api_usage_once(self) if not hasattr(backbone, "out_channels"): raise ValueError( diff --git a/torchvision/models/detection/ssd.py b/torchvision/models/detection/ssd.py index 5a068a0f0cc..2ae8709972f 100644 --- a/torchvision/models/detection/ssd.py +++ b/torchvision/models/detection/ssd.py @@ -8,6 +8,7 @@ from ..._internally_replaced_utils import load_state_dict_from_url from ...ops import boxes as box_ops +from ...utils import _log_api_usage_once from .. import vgg from . import _utils as det_utils from .anchor_utils import DefaultBoxGenerator @@ -181,6 +182,7 @@ def __init__( positive_fraction: float = 0.25, ): super().__init__() + _log_api_usage_once(self) self.backbone = backbone diff --git a/torchvision/models/detection/ssdlite.py b/torchvision/models/detection/ssdlite.py index e32aebcf839..5a7b4d14738 100644 --- a/torchvision/models/detection/ssdlite.py +++ b/torchvision/models/detection/ssdlite.py @@ -8,6 +8,7 @@ from ..._internally_replaced_utils import load_state_dict_from_url from ...ops.misc import ConvNormActivation +from ...utils import _log_api_usage_once from .. import mobilenet from . import _utils as det_utils from .anchor_utils import DefaultBoxGenerator @@ -119,6 +120,7 @@ def __init__( min_depth: int = 16, ): super().__init__() + _log_api_usage_once(self) assert not backbone[c4_pos].use_res_connect self.features = nn.Sequential(