From 84a9a9137f64c271d9e1cbc34cd022fcfa7c7c43 Mon Sep 17 00:00:00 2001 From: seinpark Date: Mon, 1 Sep 2025 15:41:15 +0900 Subject: [PATCH] fix docstring only --- mmrotate/models/roi_heads/oriented_standard_roi_head.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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. """