"...resnet50_tensorflow.git" did not exist on "bcce419adedd394ba0c04e83a02d76d84ec4b7a2"
Commit 2dbe7ba0 authored by syiming's avatar syiming
Browse files

fix docstring

parent 13078601
...@@ -843,9 +843,9 @@ class FasterRCNNMetaArch(model.DetectionModel): ...@@ -843,9 +843,9 @@ class FasterRCNNMetaArch(model.DetectionModel):
Returns: Returns:
prediction_dict: a dictionary holding "raw" prediction tensors: prediction_dict: a dictionary holding "raw" prediction tensors:
1) rpn_box_predictor_features: A 4-D float32/bfloat16 tensor with shape 1) rpn_box_predictor_features: A list of 4-D float32/bfloat16 tensor
[batch_size, height_i, width_j, depth] to be used for predicting proposal with shape [batch_size, height_i, width_j, depth] to be used for
boxes and corresponding objectness scores. predicting proposal boxes and corresponding objectness scores.
2) rpn_features_to_crop: A list of 4-D float32/bfloat16 tensor with shape 2) rpn_features_to_crop: A list of 4-D float32/bfloat16 tensor with shape
[batch_size, height, width, depth] representing image features to crop [batch_size, height, width, depth] representing image features to crop
using the proposal boxes predicted by the RPN. using the proposal boxes predicted by the RPN.
...@@ -941,8 +941,8 @@ class FasterRCNNMetaArch(model.DetectionModel): ...@@ -941,8 +941,8 @@ class FasterRCNNMetaArch(model.DetectionModel):
predictions (logits) for each of the anchors. Note that this predictions (logits) for each of the anchors. Note that this
tensor *includes* background class predictions (at class index 0). tensor *includes* background class predictions (at class index 0).
rpn_features_to_crop: A list of 4-D float32 or bfloat16 tensor with shape rpn_features_to_crop: A list of 4-D float32 or bfloat16 tensor with shape
[batch_size, height, width, depth] representing image features to crop [batch_size, height_i, width_i, depth] representing image features to
using the proposal boxes predicted by the RPN. crop using the proposal boxes predicted by the RPN.
anchors: 2-D float tensor of shape anchors: 2-D float tensor of shape
[num_anchors, self._box_coder.code_size]. [num_anchors, self._box_coder.code_size].
image_shape: A 1D int32 tensors of size [4] containing the image shape. image_shape: A 1D int32 tensors of size [4] containing the image shape.
...@@ -1006,8 +1006,8 @@ class FasterRCNNMetaArch(model.DetectionModel): ...@@ -1006,8 +1006,8 @@ class FasterRCNNMetaArch(model.DetectionModel):
Args: Args:
rpn_features_to_crop: A list 4-D float32 or bfloat16 tensor with shape rpn_features_to_crop: A list 4-D float32 or bfloat16 tensor with shape
[batch_size, height, width, depth] representing image features to crop [batch_size, height_i, width_i, depth] representing image features to
using the proposal boxes predicted by the RPN. crop using the proposal boxes predicted by the RPN.
proposal_boxes_normalized: A float tensor with shape [batch_size, proposal_boxes_normalized: A float tensor with shape [batch_size,
max_num_proposals, 4] representing the (potentially zero padded) max_num_proposals, 4] representing the (potentially zero padded)
proposal boxes for all images in the batch. These boxes are represented proposal boxes for all images in the batch. These boxes are represented
......
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