Skip to content

Commit 3ef61ee

Browse files
authored
Add support for box prompts for SamGeo2 (#340)
* Add support for box prompts for SamGeo2 * Fix numpy error
1 parent d78167a commit 3ef61ee

File tree

5 files changed

+458
-52
lines changed

5 files changed

+458
-52
lines changed

docs/examples/box_prompts.ipynb

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
"[![image](https://studiolab.sagemaker.aws/studiolab.svg)](https://studiolab.sagemaker.aws/import/github/opengeos/segment-geospatial/blob/main/docs/examples/box_prompts.ipynb)\n",
1010
"[![image](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/opengeos/segment-geospatial/blob/main/docs/examples/box_prompts.ipynb)\n",
1111
"\n",
12-
"This notebook shows how to generate object masks from text prompts with the Segment Anything Model (SAM). \n",
12+
"This notebook shows how to generate object masks from box prompts with the Segment Anything Model (SAM). \n",
1313
"\n",
1414
"Make sure you use GPU runtime for this notebook. For Google Colab, go to `Runtime` -> `Change runtime type` and select `GPU` as the hardware accelerator. "
1515
]
@@ -131,10 +131,6 @@
131131
"source": [
132132
"## Initialize SAM class\n",
133133
"\n",
134-
"The initialization of the LangSAM class might take a few minutes. The initialization downloads the model weights and sets up the model for inference.\n",
135-
"\n",
136-
"Specify the file path to the model checkpoint. If it is not specified, the model will to downloaded to the working directory.\n",
137-
"\n",
138134
"Set `automatic=False` to disable the `SamAutomaticMaskGenerator` and enable the `SamPredictor`."
139135
]
140136
},

0 commit comments

Comments
 (0)