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
145ac875
Commit
145ac875
authored
Jul 26, 2020
by
syiming
Browse files
fix coding style
parent
0e0f739b
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
62 additions
and
52 deletions
+62
-52
research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
...dels/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
+62
-52
No files found.
research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
View file @
145ac875
...
@@ -66,6 +66,16 @@ class ResnetFPN(tf.keras.layers.Layer):
...
@@ -66,6 +66,16 @@ class ResnetFPN(tf.keras.layers.Layer):
self
.
_base_fpn_max_level
=
base_fpn_max_level
self
.
_base_fpn_max_level
=
base_fpn_max_level
def
call
(
self
,
inputs
):
def
call
(
self
,
inputs
):
"""Create ResnetFPN layer.
Args:
inputs: A [batch, height_out, width_out, channels] float32 tensor
representing a batch of images.
Return:
feature_maps: A list of tensors with shape [batch, height, width, depth]
represent extracted features.
"""
inputs
=
ops
.
pad_to_multiple
(
inputs
,
self
.
pad_to_multiple
)
inputs
=
ops
.
pad_to_multiple
(
inputs
,
self
.
pad_to_multiple
)
backbone_outputs
=
self
.
classification_backbone
(
inputs
)
backbone_outputs
=
self
.
classification_backbone
(
inputs
)
...
...
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