"sgl-kernel/python/vscode:/vscode.git/clone" did not exist on "4740288303ae083e05e9ae56a8d9b068ed500e4d"
Commit 221fc6b4 authored by Yuqi Li's avatar Yuqi Li Committed by TF Object Detection Team
Browse files

nit: swap the order of `scores` and `classes` in `dummy_post_processing` when exporting tflite.

PiperOrigin-RevId: 355320914
parent c0918e55
......@@ -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, classes, num_detections
return boxes, classes, scores, num_detections
return dummy_post_processing
......
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