modify explanation

roboflow · Aug 30, 2024 · 844aec7 · 844aec7
1 parent 2dc2dfe
commit 844aec7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebooks/how-to-segment-images-with-sam-2.ipynb b/notebooks/how-to-segment-images-with-sam-2.ipynb
@@ -1085,8 +1085,8 @@
     "    multimask_output=False\n",
     ")\n",
     "\n",
-    "# A single bounding box is already a 3D np.array\n",
-    "# multiple bounding boxes need to be squeezed\n",
+    "# With one box as input, predictor returns masks of shape (1, H, W);\n",
+    "# with N boxes, it returns (N, 1, H, W).\n",
     "if boxes.shape[0] != 1:\n",
     "    masks = np.squeeze(masks)\n"
    ],