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
221fc6b4
Commit
221fc6b4
authored
Feb 02, 2021
by
Yuqi Li
Committed by
TF Object Detection Team
Feb 02, 2021
Browse files
nit: swap the order of `scores` and `classes` in `dummy_post_processing` when exporting tflite.
PiperOrigin-RevId: 355320914
parent
c0918e55
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
research/object_detection/export_tflite_graph_lib_tf2.py
research/object_detection/export_tflite_graph_lib_tf2.py
+1
-1
No files found.
research/object_detection/export_tflite_graph_lib_tf2.py
View file @
221fc6b4
...
...
@@ -145,7 +145,7 @@ class SSDModule(tf.Module):
scores
=
tf
.
constant
(
0.0
,
dtype
=
tf
.
float32
,
name
=
'scores'
)
classes
=
tf
.
constant
(
0.0
,
dtype
=
tf
.
float32
,
name
=
'classes'
)
num_detections
=
tf
.
constant
(
0.0
,
dtype
=
tf
.
float32
,
name
=
'num_detections'
)
return
boxes
,
scores
,
class
es
,
num_detections
return
boxes
,
classes
,
scor
es
,
num_detections
return
dummy_post_processing
...
...
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