Skip to content

GitLab

Commit ae3cf5e7 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

colab tutorials

parent 4f7d403d
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 6 deletions
research/object_detection/colab_tutorials/inference_from_saved_model_tf2_colab.ipynb
View file @ ae3cf5e7
...@@ -287,11 +287,6 @@ ...@@ -287,11 +287,6 @@
" end_time = time.time()\n", " end_time = time.time()\n",
" elapsed.append(end_time - start_time)\n", " elapsed.append(end_time - start_time)\n",
"\n", "\n",
" # detection_masks_reframed = ops.reframe_box_masks_to_image_masks(\n",
" # detections['detection_masks'][0],\n",
" # detections['detection_boxes'][0], image_np.shape[0], image_np.shape[1])\n",
" # detection_masks_reframed = tf.cast(detection_masks_reframed > 0.5,\n",
" # tf.uint8).numpy()\n",
" plt.rcParams['figure.figsize'] = [42, 21]\n", " plt.rcParams['figure.figsize'] = [42, 21]\n",
" label_id_offset = 1\n", " label_id_offset = 1\n",
" image_np_with_detections = image_np.copy()\n", " image_np_with_detections = image_np.copy()\n",
...@@ -301,7 +296,6 @@ ...@@ -301,7 +296,6 @@
" detections['detection_classes'][0].numpy().astype(np.int32),\n", " detections['detection_classes'][0].numpy().astype(np.int32),\n",
" detections['detection_scores'][0].numpy(),\n", " detections['detection_scores'][0].numpy(),\n",
" category_index,\n", " category_index,\n",
" # instance_masks=detection_masks_reframed,\n",
" use_normalized_coordinates=True,\n", " use_normalized_coordinates=True,\n",
" max_boxes_to_draw=200,\n", " max_boxes_to_draw=200,\n",
" min_score_thresh=.40,\n", " min_score_thresh=.40,\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