From 844aec7ff0b737d217da7d934030892325a7a477 Mon Sep 17 00:00:00 2001 From: venkatram-dev Date: Fri, 30 Aug 2024 08:15:46 -0700 Subject: [PATCH] modify explanation --- notebooks/how-to-segment-images-with-sam-2.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/notebooks/how-to-segment-images-with-sam-2.ipynb b/notebooks/how-to-segment-images-with-sam-2.ipynb index c40ffa7..7afd10b 100644 --- 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" ],