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
0d52af26
Commit
0d52af26
authored
Aug 22, 2022
by
A. Unique TensorFlower
Browse files
Internal change
PiperOrigin-RevId: 469212225
parent
97a62d36
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
official/projects/edgetpu/vision/modeling/heads/bifpn_head.py
...cial/projects/edgetpu/vision/modeling/heads/bifpn_head.py
+2
-2
No files found.
official/projects/edgetpu/vision/modeling/heads/bifpn_head.py
View file @
0d52af26
...
@@ -670,7 +670,7 @@ class SegClassNet(tf.keras.layers.Layer):
...
@@ -670,7 +670,7 @@ class SegClassNet(tf.keras.layers.Layer):
self
.
min_level
=
min_level
self
.
min_level
=
min_level
self
.
max_level
=
max_level
self
.
max_level
=
max_level
self
.
fullres_output
=
fullres_output
self
.
fullres_output
=
fullres_output
self
.
fullres_
conv_transpose
=
fullres_skip_connections
self
.
fullres_
skip_connections
=
fullres_skip_connections
self
.
fnode
=
FNode
(
self
.
fnode
=
FNode
(
0
,
# Always use the first level with highest resolution.
0
,
# Always use the first level with highest resolution.
...
@@ -726,7 +726,7 @@ class SegClassNet(tf.keras.layers.Layer):
...
@@ -726,7 +726,7 @@ class SegClassNet(tf.keras.layers.Layer):
if
self
.
fullres_output
:
if
self
.
fullres_output
:
for
i
in
reversed
(
range
(
self
.
min_level
)):
for
i
in
reversed
(
range
(
self
.
min_level
)):
if
self
.
config
.
fullres_skip_connections
:
if
self
.
fullres_skip_connections
:
net
=
tf
.
keras
.
layers
.
Concatenate
()([
net
,
backbone_feats
[
i
+
1
]])
net
=
tf
.
keras
.
layers
.
Concatenate
()([
net
,
backbone_feats
[
i
+
1
]])
net
=
self
.
fullres_conv
[
str
(
i
)](
net
)
net
=
self
.
fullres_conv
[
str
(
i
)](
net
)
net
=
self
.
fullres_conv_transpose
[
str
(
i
)](
net
)
net
=
self
.
fullres_conv_transpose
[
str
(
i
)](
net
)
...
...
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