change doc string for rpn_box_predictor_features

4dcc11d1 · syiming · e186a940 · 4dcc11d1
Commit 4dcc11d1 authored Jul 29, 2020 by syiming
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

research/object_detection/meta_architectures/faster_rcnn_meta_arch.py ...ect_detection/meta_architectures/faster_rcnn_meta_arch.py +2 -2

No files found.
--- a/research/object_detection/meta_architectures/faster_rcnn_meta_arch.py
+++ b/research/object_detection/meta_architectures/faster_rcnn_meta_arch.py
@@ -1299,7 +1299,7 @@ class FasterRCNNMetaArch(model.DetectionModel):

    Returns:
      rpn_box_predictor_features: A list of 4-D float32 tensor with shape
-        [batch, height, width, depth] to be used for predicting proposal boxes
+        [batch, height_i, width_j, depth] to be used for predicting proposal boxes
        and corresponding objectness scores.
      rpn_features_to_crop: A list of 4-D float32 tensor with shape
        [batch, height, width, depth] representing image features to crop using
@@ -1354,7 +1354,7 @@ class FasterRCNNMetaArch(model.DetectionModel):

    Args:
      rpn_box_predictor_features: A list of 4-D float32 tensor with shape
-        [batch, height, width, depth] to be used for predicting proposal boxes
+        [batch, height_i, width_j, depth] to be used for predicting proposal boxes
        and corresponding objectness scores.

    Returns: