diff --git a/mmrotate/models/roi_heads/oriented_standard_roi_head.py b/mmrotate/models/roi_heads/oriented_standard_roi_head.py index 5f0058b7c..220c53d89 100644 --- a/mmrotate/models/roi_heads/oriented_standard_roi_head.py +++ b/mmrotate/models/roi_heads/oriented_standard_roi_head.py @@ -142,8 +142,8 @@ def simple_test_bboxes(self, Returns: tuple[list[Tensor], list[Tensor]]: The first list contains \ the boxes of the corresponding image in a batch, each \ - tensor has the shape (num_boxes, 5) and last dimension \ - 5 represent (cx, cy, w, h, a, score). Each Tensor \ + tensor has the shape (num_boxes, 6) and last dimension \ + 6 represent (cx, cy, w, h, a, score). Each Tensor \ in the second list is the labels with shape (num_boxes, ). \ The length of both lists should be equal to batch_size. """