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
d2deafd6
Commit
d2deafd6
authored
Jun 29, 2020
by
syiming
Browse files
change features_to_crop as a single tensor to make sure it is consistent with the calling funciton
parent
d41fc8f4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
research/object_detection/meta_architectures/faster_rcnn_meta_arch.py
...ect_detection/meta_architectures/faster_rcnn_meta_arch.py
+2
-1
No files found.
research/object_detection/meta_architectures/faster_rcnn_meta_arch.py
View file @
d2deafd6
...
...
@@ -1963,7 +1963,7 @@ class FasterRCNNMetaArch(model.DetectionModel):
second stage box classifier for each proposal.
Args:
features_to_crop: A
list of
float32 tensor with shape
features_to_crop: A float32 tensor with shape
[batch_size, height, width, depth]
proposal_boxes_normalized: A float32 tensor with shape [batch_size,
num_proposals, box_code_size] containing proposal boxes in
...
...
@@ -1973,6 +1973,7 @@ class FasterRCNNMetaArch(model.DetectionModel):
Returns:
A float32 tensor with shape [K, new_height, new_width, depth].
"""
features_to_crop
=
[
features_to_crop
]
num_levels
=
len
(
features_to_crop
)
box_levels
=
None
if
num_levels
!=
1
:
...
...
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