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

Update export_tflite_ssd_graph.py

parent 15b33360
......@@ -40,10 +40,10 @@ If add_postprocessing_op is true: frozen graph adds a
TFLite_Detection_PostProcess custom op node has four outputs:
detection_boxes: a float32 tensor of shape [1, num_boxes, 4] with box
locations
detection_scores: a float32 tensor of shape [1, num_boxes]
with class scores
detection_classes: a float32 tensor of shape [1, num_boxes]
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
else:
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