Expand usage logging to detection models (#4736)

* Expand usage logging to detection models

Summary: Track all detection model instantiation.

Reviewed By: sallysyw

Differential Revision: D31741603

fbshipit-source-id: 89467db5d89ea41c5a9ab95d1846264d0a1f4199

* Fix lint

Co-authored-by: Kai Zhang <[email protected]>

Author: fmassa

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

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(

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
 

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(