Improved text prompt notebook (#79)

Author: giswqs

docs/examples/text_prompts.ipynb

@@ -189,7 +189,9 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Visualize the results"
+    "## Visualize the results\n",
+    "\n",
+    "Show the result with bounding boxes on the map."
    ]
   },
   {
@@ -202,6 +204,63 @@
     "    cmap='Greens', \n",
     "    box_color='red', \n",
     "    title='Automatic Segmentation of Trees', \n",
+    "    blend=True,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "![](https://i.imgur.com/ytKMTlA.png)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Show the result without bounding boxes on the map."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "sam.show_anns(\n",
+    "    cmap='Greens', \n",
+    "    add_boxes=False,\n",
+    "    alpha=0.5,\n",
+    "    title='Automatic Segmentation of Trees', \n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "![](https://i.imgur.com/3Iq2kt1.png)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Show the result as a grayscale image."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "sam.show_anns(\n",
+    "    cmap='Greys_r', \n",
+    "    add_boxes=False,\n",
+    "    alpha=1,\n",
+    "    title='Automatic Segmentation of Trees', \n",
     "    blend=False,\n",
     "    output='trees.tif'\n",
     ")"
@@ -211,7 +270,47 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "![](https://i.imgur.com/ytKMTlA.png)"
+    "![](https://i.imgur.com/KtHwFbF.png)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Convert the result to a vector format.  "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "sam.raster_to_vector(\"trees.tif\", \"trees.shp\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Show the results on the interactive map."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "m.add_raster(\"trees.tif\", layer_name=\"Trees\", palette=\"Greens\", opacity=0.5, nodata=0)\n",
+    "style = {\n",
+    "    \"color\": \"#3388ff\",\n",
+    "    \"weight\": 2,\n",
+    "    \"fillColor\": \"#7c4185\",\n",
+    "    \"fillOpacity\": 0.5,\n",
+    "}\n",
+    "m.add_vector(\"trees.shp\", layer_name=\"Vector\", style=style)\n",
+    "m"
    ]
   }
  ],

samgeo/samgeo.py

@@ -604,6 +604,20 @@ def download_tms_as_tiff(self, source, pt1, pt2, zoom, dist):
         image = draw_tile(source, pt1[0], pt1[1], pt2[0], pt2[1], zoom, dist)
         return image
 
+    def raster_to_vector(self, image, output, simplify_tolerance=None, **kwargs):
+        """Save the result to a vector file.
+
+        Args:
+            image (str): The path to the image file.
+            output (str): The path to the vector file.
+            simplify_tolerance (float, optional): The maximum allowed geometry displacement.
+                The higher this value, the smaller the number of vertices in the resulting geometry.
+        """
+
+        raster_to_vector(
+            image, output, simplify_tolerance=simplify_tolerance, **kwargs
+        )
+
     def tiff_to_vector(self, tiff_path, output, simplify_tolerance=None, **kwargs):
         """Convert a tiff file to a gpkg file.
 

samgeo/text_sam.py

@@ -376,6 +376,20 @@ def show_anns(
             else:
                 array_to_image(self.prediction, output, self.source)
 
+    def raster_to_vector(self, image, output, simplify_tolerance=None, **kwargs):
+        """Save the result to a vector file.
+
+        Args:
+            image (str): The path to the image file.
+            output (str): The path to the vector file.
+            simplify_tolerance (float, optional): The maximum allowed geometry displacement.
+                The higher this value, the smaller the number of vertices in the resulting geometry.
+        """
+
+        raster_to_vector(
+            image, output, simplify_tolerance=simplify_tolerance, **kwargs
+        )
+
 
 def main():
     parser = argparse.ArgumentParser(description="LangSAM")