[fbsync] Add typing annotations to detection/generalized_rcnn (#4631)

Summary:
* Update typing

* Fix bug

* Unblock mypy

* Ignore small error

Reviewed By: kazhang

Differential Revision: D32216673

fbshipit-source-id: cc73b84b830c1e0c31502a2cb6cd7408a899d6ba

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

Author: 2 people

mypy.ini

@@ -46,10 +46,6 @@ ignore_errors = True
 
 ignore_errors = True
 
-[mypy-torchvision.models.detection.generalized_rcnn]
-
-ignore_errors = True
-
 [mypy-torchvision.models.detection.faster_rcnn]
 
 ignore_errors = True

torchvision/models/detection/generalized_rcnn.py

@@ -25,7 +25,7 @@ class GeneralizedRCNN(nn.Module):
             the model
     """
 
-    def __init__(self, backbone, rpn, roi_heads, transform):
+    def __init__(self, backbone: nn.Module, rpn: nn.Module, roi_heads: nn.Module, transform: nn.Module) -> None:
         super().__init__()
         _log_api_usage_once(self)
         self.transform = transform
@@ -48,7 +48,7 @@ def forward(self, images, targets=None):
         """
         Args:
             images (list[Tensor]): images to be processed
-            targets (list[Dict[Tensor]]): ground-truth boxes present in the image (optional)
+            targets (list[Dict[str, Tensor]]): ground-truth boxes present in the image (optional)
 
         Returns:
             result (list[BoxList] or dict[Tensor]): the output from the model.
@@ -97,7 +97,7 @@ def forward(self, images, targets=None):
             features = OrderedDict([("0", features)])
         proposals, proposal_losses = self.rpn(images, features, targets)
         detections, detector_losses = self.roi_heads(features, proposals, images.image_sizes, targets)
-        detections = self.transform.postprocess(detections, images.image_sizes, original_image_sizes)
+        detections = self.transform.postprocess(detections, images.image_sizes, original_image_sizes)  # type: ignore[operator]
 
         losses = {}
         losses.update(detector_losses)