Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
2dbe7ba0
Commit
2dbe7ba0
authored
Aug 05, 2020
by
syiming
Browse files
fix docstring
parent
13078601
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
7 deletions
+7
-7
research/object_detection/meta_architectures/faster_rcnn_meta_arch.py
...ect_detection/meta_architectures/faster_rcnn_meta_arch.py
+7
-7
No files found.
research/object_detection/meta_architectures/faster_rcnn_meta_arch.py
View file @
2dbe7ba0
...
@@ -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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment