polish docs (open-mmlab#6623)

ZwwWayne · Jul 18, 2022 · db0fb57 · db0fb57
1 parent b25ce2d
commit db0fb57
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 12 deletions.
diff --git a/mmdet/models/dense_heads/base_dense_head.py b/mmdet/models/dense_heads/base_dense_head.py
@@ -251,7 +251,7 @@ def _bbox_post_process(self,
         """bbox post-processing method.
 
         The boxes would be rescaled to the original image scale and do
-        the nms operation. Usually with_nms is False is used for aug test.
+        the nms operation. Usually `with_nms` is False is used for aug test.
 
         Args:
             results (obj: InstanceData): Detection instance results,

diff --git a/mmdet/models/dense_heads/rpn_head.py b/mmdet/models/dense_heads/rpn_head.py
@@ -191,13 +191,12 @@ def _bbox_post_process(self, mlvl_scores, mlvl_bboxes, mlvl_valid_anchors,
                            level_ids, cfg, img_shape, img_meta, **kwargs):
         """bbox post-processing method.
 
-        The boxes would be rescaled to the original image scale and do
-        the nms operation. Usually with_nms is False is used for aug test.
+        Do the nms operation for bboxes in same level.
 
         Args:
             mlvl_scores (list[Tensor]): Box scores from all scale
                 levels of a single image, each item has shape
-                (num_bboxes, num_class).
+                (num_bboxes, ).
             mlvl_bboxes (list[Tensor]): Decoded bboxes from all scale
                 levels of a single image, each item has shape (num_bboxes, 4).
             mlvl_valid_anchors (list[Tensor]): Anchors of all scale level

diff --git a/mmdet/models/roi_heads/bbox_heads/bbox_head.py b/mmdet/models/roi_heads/bbox_heads/bbox_head.py
@@ -131,11 +131,12 @@ def _get_target_single(self, pos_bboxes, neg_bboxes, pos_gt_bboxes,
             neg_bboxes (Tensor): Contains all the negative boxes,
                 has shape (num_neg, 4), the last dimension 4
                 represents [tl_x, tl_y, br_x, br_y].
-            pos_gt_bboxes (Tensor): Contains all the gt_boxes,
-                has shape (num_gt, 4), the last dimension 4
+            pos_gt_bboxes (Tensor): Contains gt_boxes for
+                all positive samples, has shape (num_pos, 4),
+                the last dimension 4
                 represents [tl_x, tl_y, br_x, br_y].
-            pos_gt_labels (Tensor): Contains all the gt_labels,
-                has shape (num_gt).
+            pos_gt_labels (Tensor): Contains gt_labels for
+                all positive samples, has shape (num_pos, ).
             cfg (obj:`ConfigDict`): `train_cfg` of R-CNN.
 
         Returns:

diff --git a/mmdet/models/roi_heads/bbox_heads/dii_head.py b/mmdet/models/roi_heads/bbox_heads/dii_head.py
@@ -305,11 +305,12 @@ def _get_target_single(self, pos_inds, neg_inds, pos_bboxes, neg_bboxes,
             neg_bboxes (Tensor): Contains all the negative boxes,
                 has shape (num_neg, 4), the last dimension 4
                 represents [tl_x, tl_y, br_x, br_y].
-            pos_gt_bboxes (Tensor): Contains all the gt_boxes,
-                has shape (num_gt, 4), the last dimension 4
+            pos_gt_bboxes (Tensor): Contains gt_boxes for
+                all positive samples, has shape (num_pos, 4),
+                the last dimension 4
                 represents [tl_x, tl_y, br_x, br_y].
-            pos_gt_labels (Tensor): Contains all the gt_labels,
-                has shape (num_gt).
+            pos_gt_labels (Tensor): Contains gt_labels for
+                all positive samples, has shape (num_pos, ).
             cfg (obj:`ConfigDict`): `train_cfg` of R-CNN.
 
         Returns: