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