Unverified Commit f28d93b0 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

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: default avatarKai Zhang <kaizh@fb.com>
parent 5c57f5ec
...@@ -9,6 +9,8 @@ from typing import Tuple, List, Dict, Optional, Union ...@@ -9,6 +9,8 @@ from typing import Tuple, List, Dict, Optional, Union
import torch import torch
from torch import nn, Tensor from torch import nn, Tensor
from ...utils import _log_api_usage_once
class GeneralizedRCNN(nn.Module): class GeneralizedRCNN(nn.Module):
""" """
...@@ -25,6 +27,7 @@ class GeneralizedRCNN(nn.Module): ...@@ -25,6 +27,7 @@ class GeneralizedRCNN(nn.Module):
def __init__(self, backbone, rpn, roi_heads, transform): def __init__(self, backbone, rpn, roi_heads, transform):
super(GeneralizedRCNN, self).__init__() super(GeneralizedRCNN, self).__init__()
_log_api_usage_once(self)
self.transform = transform self.transform = transform
self.backbone = backbone self.backbone = backbone
self.rpn = rpn self.rpn = rpn
......
...@@ -11,6 +11,7 @@ from ...ops import sigmoid_focal_loss ...@@ -11,6 +11,7 @@ from ...ops import sigmoid_focal_loss
from ...ops import boxes as box_ops from ...ops import boxes as box_ops
from ...ops import misc as misc_nn_ops from ...ops import misc as misc_nn_ops
from ...ops.feature_pyramid_network import LastLevelP6P7 from ...ops.feature_pyramid_network import LastLevelP6P7
from ...utils import _log_api_usage_once
from ..resnet import resnet50 from ..resnet import resnet50
from . import _utils as det_utils from . import _utils as det_utils
from ._utils import overwrite_eps from ._utils import overwrite_eps
...@@ -336,6 +337,7 @@ class RetinaNet(nn.Module): ...@@ -336,6 +337,7 @@ class RetinaNet(nn.Module):
topk_candidates=1000, topk_candidates=1000,
): ):
super().__init__() super().__init__()
_log_api_usage_once(self)
if not hasattr(backbone, "out_channels"): if not hasattr(backbone, "out_channels"):
raise ValueError( raise ValueError(
......
...@@ -8,6 +8,7 @@ from torch import nn, Tensor ...@@ -8,6 +8,7 @@ from torch import nn, Tensor
from ..._internally_replaced_utils import load_state_dict_from_url from ..._internally_replaced_utils import load_state_dict_from_url
from ...ops import boxes as box_ops from ...ops import boxes as box_ops
from ...utils import _log_api_usage_once
from .. import vgg from .. import vgg
from . import _utils as det_utils from . import _utils as det_utils
from .anchor_utils import DefaultBoxGenerator from .anchor_utils import DefaultBoxGenerator
...@@ -181,6 +182,7 @@ class SSD(nn.Module): ...@@ -181,6 +182,7 @@ class SSD(nn.Module):
positive_fraction: float = 0.25, positive_fraction: float = 0.25,
): ):
super().__init__() super().__init__()
_log_api_usage_once(self)
self.backbone = backbone self.backbone = backbone
......
...@@ -8,6 +8,7 @@ from torch import nn, Tensor ...@@ -8,6 +8,7 @@ from torch import nn, Tensor
from ..._internally_replaced_utils import load_state_dict_from_url from ..._internally_replaced_utils import load_state_dict_from_url
from ...ops.misc import ConvNormActivation from ...ops.misc import ConvNormActivation
from ...utils import _log_api_usage_once
from .. import mobilenet from .. import mobilenet
from . import _utils as det_utils from . import _utils as det_utils
from .anchor_utils import DefaultBoxGenerator from .anchor_utils import DefaultBoxGenerator
...@@ -119,6 +120,7 @@ class SSDLiteFeatureExtractorMobileNet(nn.Module): ...@@ -119,6 +120,7 @@ class SSDLiteFeatureExtractorMobileNet(nn.Module):
min_depth: int = 16, min_depth: int = 16,
): ):
super().__init__() super().__init__()
_log_api_usage_once(self)
assert not backbone[c4_pos].use_res_connect assert not backbone[c4_pos].use_res_connect
self.features = nn.Sequential( self.features = nn.Sequential(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment