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
8dba5924
Unverified
Commit
8dba5924
authored
Jul 20, 2018
by
achowdhery
Committed by
GitHub
Jul 20, 2018
Browse files
Update export_tflite_ssd_graph_lib.py
Correcting comments
parent
220b7e0c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
research/object_detection/export_tflite_ssd_graph_lib.py
research/object_detection/export_tflite_ssd_graph_lib.py
+3
-3
No files found.
research/object_detection/export_tflite_ssd_graph_lib.py
View file @
8dba5924
...
@@ -41,7 +41,7 @@ def get_const_center_size_encoded_anchors(anchors):
...
@@ -41,7 +41,7 @@ def get_const_center_size_encoded_anchors(anchors):
boxes
boxes
Returns:
Returns:
encoded_anchors: a float32 constant tensor of shape [
4,
num_anchors]
encoded_anchors: a float32 constant tensor of shape [num_anchors
, 4
]
containing the anchor boxes.
containing the anchor boxes.
"""
"""
anchor_boxlist
=
box_list
.
BoxList
(
anchors
)
anchor_boxlist
=
box_list
.
BoxList
(
anchors
)
...
@@ -83,10 +83,10 @@ def append_postprocessing_op(frozen_graph_def, max_detections,
...
@@ -83,10 +83,10 @@ def append_postprocessing_op(frozen_graph_def, max_detections,
TFLite_Detection_PostProcess custom op node has four outputs:
TFLite_Detection_PostProcess custom op node has four outputs:
detection_boxes: a float32 tensor of shape [1, num_boxes, 4] with box
detection_boxes: a float32 tensor of shape [1, num_boxes, 4] with box
locations
locations
detection_scores: a float32 tensor of shape [1, num_boxes]
with class scores
detection_classes: a float32 tensor of shape [1, num_boxes]
detection_classes: a float32 tensor of shape [1, num_boxes]
with class indices
with class indices
detection_scores: a float32 tensor of shape [1, num_boxes]
with class scores
num_boxes: a float32 tensor of size 1 containing the number of detected
num_boxes: a float32 tensor of size 1 containing the number of detected
boxes
boxes
"""
"""
...
...
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