Merge pull request #1 from naeem-md93/naeem-md93-patch-1

Update roi_heads.py

Author: naeem-md93

torchvision/models/detection/roi_heads.py

@@ -826,11 +826,7 @@ def forward(
 
         # keep none checks in if conditional so torchscript will conditionally
         # compile each branch
-        if (
-            self.keypoint_roi_pool is not None
-            and self.keypoint_head is not None
-            and self.keypoint_predictor is not None
-        ):
+        if self.has_keypoint():
             keypoint_proposals = [p["boxes"] for p in result]
             if self.training:
                 # during training, only focus on positive boxes