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
Show 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 @@ ...@@ -244,10 +244,10 @@
"\r\n", "\r\n",
" interpreter.invoke()\r\n", " interpreter.invoke()\r\n",
"\r\n", "\r\n",
" boxes = interpreter.get_tensor(output_details[0]['index'])\r\n", " scores = interpreter.get_tensor(output_details[0]['index'])\r\n",
" classes = interpreter.get_tensor(output_details[1]['index'])\r\n", " boxes = 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[2]['index'])\r\n",
" num_detections = interpreter.get_tensor(output_details[3]['index'])\r\n", " classes = interpreter.get_tensor(output_details[3]['index'])\r\n",
"\r\n", "\r\n",
" if include_keypoint:\r\n", " if include_keypoint:\r\n",
" kpts = interpreter.get_tensor(output_details[4]['index'])\r\n", " kpts = interpreter.get_tensor(output_details[4]['index'])\r\n",
......
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