Add support for box prompts for SamGeo2 (#340)

* Add support for box prompts for SamGeo2 * Fix numpy error
opengeos · Oct 14, 2024 · 3ef61ee · 3ef61ee
1 parent d78167a
commit 3ef61ee
Show file tree

Hide file tree

Showing 5 changed files with 458 additions and 52 deletions.
diff --git a/docs/examples/box_prompts.ipynb b/docs/examples/box_prompts.ipynb
@@ -9,7 +9,7 @@
     "[![image](https://studiolab.sagemaker.aws/studiolab.svg)](https://studiolab.sagemaker.aws/import/github/opengeos/segment-geospatial/blob/main/docs/examples/box_prompts.ipynb)\n",
     "[![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",
     "\n",
-    "This notebook shows how to generate object masks from text prompts with the Segment Anything Model (SAM). \n",
+    "This notebook shows how to generate object masks from box prompts with the Segment Anything Model (SAM). \n",
     "\n",
     "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. "
    ]
@@ -131,10 +131,6 @@
    "source": [
     "## Initialize SAM class\n",
     "\n",
-    "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",
-    "\n",
-    "Specify the file path to the model checkpoint. If it is not specified, the model will to downloaded to the working directory.\n",
-    "\n",
     "Set `automatic=False` to disable the `SamAutomaticMaskGenerator` and enable the `SamPredictor`."
    ]
   },