[Fix] 'RoIAlignRotated' object has no attribute 'output_size' (#213)

* Update rotate_single_level_roi_extractor.py * Update rotate_single_level_roi_extractor.py
open-mmlab · Apr 20, 2022 · 382c06d · 382c06d
1 parent df125d7
commit 382c06d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mmrotate/models/roi_heads/roi_extractors/rotate_single_level_roi_extractor.py b/mmrotate/models/roi_heads/roi_extractors/rotate_single_level_roi_extractor.py
@@ -99,7 +99,9 @@ def forward(self, feats, rois, roi_scale_factor=None):
         Returns:
             torch.Tensor: Scaled RoI features.
         """
-        if isinstance(self.roi_layers[0], ops.RiRoIAlignRotated):
+        from mmrotate import digit_version, mmcv_version
+        if isinstance(self.roi_layers[0], ops.RiRoIAlignRotated
+                      ) or mmcv_version == digit_version('1.4.5'):
             out_size = nn.modules.utils._pair(self.roi_layers[0].out_size)
         else:
             out_size = self.roi_layers[0].output_size