Skip to content

GitLab

Commit a9c5469d authored by TF Object Detection Team's avatar TF Object Detection Team
Browse files

Merge pull request #10223 from airman00:patch-2 

PiperOrigin-RevId: 395336069
parents 18d62791 843111ca
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
research/object_detection/colab_tutorials/centernet_on_device.ipynb
View file @ a9c5469d
......@@ -244,10 +244,10 @@
"\r\n",
" interpreter.invoke()\r\n",
"\r\n",
" boxes = interpreter.get_tensor(output_details[0]['index'])\r\n",
" classes = interpreter.get_tensor(output_details[1]['index'])\r\n",
" scores = interpreter.get_tensor(output_details[2]['index'])\r\n",
" num_detections = interpreter.get_tensor(output_details[3]['index'])\r\n",
" scores = interpreter.get_tensor(output_details[0]['index'])\r\n",
" boxes = interpreter.get_tensor(output_details[1]['index'])\r\n",
" num_detections = interpreter.get_tensor(output_details[2]['index'])\r\n",
" classes = interpreter.get_tensor(output_details[3]['index'])\r\n",
"\r\n",
" if include_keypoint:\r\n",
" kpts = interpreter.get_tensor(output_details[4]['index'])\r\n",
......@@ -759,4 +759,4 @@
]
}
]
}
\ No newline at end of file
}
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