add logging for LRASPP, FCN, DeepPLabV3 (#4956)

Co-authored-by: Vasilis Vryniotis <[email protected]>

Author: kazhang

torchvision/models/segmentation/_utils.py

@@ -5,13 +5,15 @@
 from torch.nn import functional as F
 
 from ..._internally_replaced_utils import load_state_dict_from_url
+from ...utils import _log_api_usage_once
 
 
 class _SimpleSegmentationModel(nn.Module):
     __constants__ = ["aux_classifier"]
 
     def __init__(self, backbone: nn.Module, classifier: nn.Module, aux_classifier: Optional[nn.Module] = None) -> None:
         super().__init__()
+        _log_api_usage_once(self)
         self.backbone = backbone
         self.classifier = classifier
         self.aux_classifier = aux_classifier

torchvision/models/segmentation/lraspp.py

@@ -4,6 +4,7 @@
 from torch import nn, Tensor
 from torch.nn import functional as F
 
+from ...utils import _log_api_usage_once
 from .. import mobilenetv3
 from ..feature_extraction import create_feature_extractor
 from ._utils import _load_weights
@@ -37,6 +38,7 @@ def __init__(
         self, backbone: nn.Module, low_channels: int, high_channels: int, num_classes: int, inter_channels: int = 128
     ) -> None:
         super().__init__()
+        _log_api_usage_once(self)
         self.backbone = backbone
         self.classifier = LRASPPHead(low_channels, high_channels, num_classes, inter_channels)