Skip to content

GitLab

Commit 4dcc11d1 authored by syiming's avatar syiming
Browse files

change doc string for rpn_box_predictor_features

parent e186a940
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/object_detection/meta_architectures/faster_rcnn_meta_arch.py
View file @ 4dcc11d1
...@@ -1299,7 +1299,7 @@ class FasterRCNNMetaArch(model.DetectionModel): ...@@ -1299,7 +1299,7 @@ class FasterRCNNMetaArch(model.DetectionModel):
Returns: Returns:
rpn_box_predictor_features: A list of 4-D float32 tensor with shape 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. and corresponding objectness scores.
rpn_features_to_crop: A list of 4-D float32 tensor with shape rpn_features_to_crop: A list of 4-D float32 tensor with shape
[batch, height, width, depth] representing image features to crop using [batch, height, width, depth] representing image features to crop using
...@@ -1354,7 +1354,7 @@ class FasterRCNNMetaArch(model.DetectionModel): ...@@ -1354,7 +1354,7 @@ class FasterRCNNMetaArch(model.DetectionModel):
Args: Args:
rpn_box_predictor_features: A list of 4-D float32 tensor with shape 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. and corresponding objectness scores.
Returns: Returns:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment