Skip to content

GitLab

Unverified Commit 220b7e0c authored by achowdhery's avatar achowdhery Committed by GitHub
Browse files

Update export_tflite_ssd_graph.py

parent 15b33360
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/object_detection/export_tflite_ssd_graph.py
View file @ 220b7e0c
...@@ -40,10 +40,10 @@ If add_postprocessing_op is true: frozen graph adds a ...@@ -40,10 +40,10 @@ If add_postprocessing_op is true: frozen graph adds a
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 boxes num_boxes: a float32 tensor of size 1 containing the number of detected boxes
else: else:
the graph has two outputs: the graph has two outputs:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment