Avoid class index overflow when using OIDv4 labels (#6641)

np.uint8 to np.int64 for detection class label index overflow when using OIDv4 labels

Avoid class index overflow when using OIDv4 labels (#6641)
np.uint8 to np.int64 for detection class label index overflow when using OIDv4 labels
c96422f7 · Akash Paul · pkulzc · 9d299984 · c96422f7
Commit c96422f7 authored Apr 24, 2019 by Akash Paul Committed by pkulzc Apr 23, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

research/object_detection/object_detection_tutorial.ipynb research/object_detection/object_detection_tutorial.ipynb +1 -1

No files found.
--- a/research/object_detection/object_detection_tutorial.ipynb
+++ b/research/object_detection/object_detection_tutorial.ipynb
@@ -387,7 +387,7 @@
        "      # all outputs are float32 numpy arrays, so convert types as appropriate\n",
        "      output_dict['num_detections'] = int(output_dict['num_detections'][0])\n",
        "      output_dict['detection_classes'] = output_dict[\n",
-        "          'detection_classes'][0].astype(np.uint8)\n",
+        "          'detection_classes'][0].astype(np.int64)\n",
        "      output_dict['detection_boxes'] = output_dict['detection_boxes'][0]\n",
        "      output_dict['detection_scores'] = output_dict['detection_scores'][0]\n",
        "      if 'detection_masks' in output_dict:\n",