diff --git a/mmdet/models/detectors/cascade_rcnn.py b/mmdet/models/detectors/cascade_rcnn.py
index 3e2d1afaddf19b1d48a265e220a89b3474bb9e22..86e971b8c57cba0dd958f9ae8a445d54498319f1 100644
--- a/mmdet/models/detectors/cascade_rcnn.py
+++ b/mmdet/models/detectors/cascade_rcnn.py
@@ -159,7 +159,7 @@ class CascadeRCNN(BaseDetector, RPNTestMixin):
                       proposals=None):
         """
         Args:
-            img (Tensor): of shape (B, C, H, W) encoding input images.
+            img (Tensor): of shape (N, C, H, W) encoding input images.
                 Typically these should be mean centered and std scaled.
 
             img_meta (list[dict]): list of image info dict where each dict has:
@@ -301,7 +301,7 @@ class CascadeRCNN(BaseDetector, RPNTestMixin):
         """Run inference on a single image.
 
         Args:
-            img (Tensor): must be in shape (B, C, H, W)
+            img (Tensor): must be in shape (N, C, H, W)
             img_meta (list[dict]): a list with one dictionary element.
                 See `mmdet/datasets/pipelines/formatting.py:Collect` for
                 details of meta dicts.
diff --git a/mmdet/models/detectors/two_stage.py b/mmdet/models/detectors/two_stage.py
index f35d25b2455fe91f0d2bbb2a2ea13680b2986c70..e457ea31d8301ed0e10e535f65e53968274b5fc6 100644
--- a/mmdet/models/detectors/two_stage.py
+++ b/mmdet/models/detectors/two_stage.py
@@ -137,7 +137,7 @@ class TwoStageDetector(BaseDetector, RPNTestMixin, BBoxTestMixin,
                       proposals=None):
         """
         Args:
-            img (Tensor): of shape (B, C, H, W) encoding input images.
+            img (Tensor): of shape (N, C, H, W) encoding input images.
                 Typically these should be mean centered and std scaled.
 
             img_meta (list[dict]): list of image info dict where each dict has: