geoai-py 0.10.0__py2.py3-none-any.whl → 0.11.1__py2.py3-none-any.whl

geoai/geoai.py

@@ -41,6 +41,7 @@ from .train import (
     train_segmentation_model,
 )
 from .utils import *
+from .map_widgets import DINOv3GUI
 
 
 class Map(leafmap.Map):
@@ -50,6 +51,16 @@ class Map(leafmap.Map):
         """Initialize the Map class."""
         super().__init__(*args, **kwargs)
 
+    def add_dinov3_gui(
+        self,
+        raster: str,
+        processor: "DINOv3GeoProcessor",
+        features: torch.Tensor,
+        **kwargs: Any,
+    ) -> None:
+        """Add a DINOv3 GUI to the map."""
+        return DINOv3GUI(raster, processor, features, host_map=self, **kwargs)
+
 
 class MapLibre(maplibregl.Map):
     """A subclass of maplibregl.Map for GeoAI applications."""

geoai/map_widgets.py

@@ -0,0 +1,174 @@
+"""Interactive widget for GeoAI."""
+
+import ipywidgets as widgets
+from .utils import dict_to_image, dict_to_rioxarray
+
+
+class DINOv3GUI(widgets.VBox):
+    """Interactive widget for DINOv3."""
+
+    def __init__(
+        self,
+        raster: str,
+        processor=None,
+        features=None,
+        host_map=None,
+        position="topright",
+        colormap_options=None,
+        raster_args=None,
+    ):
+        """Initialize the DINOv3 GUI.
+
+        Args:
+            raster (str): The path to the raster image.
+            processor (DINOv3GeoProcessor): The DINOv3 processor.
+            features (torch.Tensor): The features of the raster image.
+            host_map (Map): The host map.
+            position (str): The position of the widget.
+            colormap_options (list): The colormap options.
+            raster_args (dict): The raster arguments.
+
+        Returns:
+            None
+
+        Example:
+            >>> processor = DINOv3GeoProcessor()
+            >>> features, h_patches, w_patches = processor.extract_features(raster)
+            >>> gui = DINOv3GUI(raster, processor, features, host_map=m)
+        """
+        if raster_args is None:
+            raster_args = {}
+
+        if "layer_name" not in raster_args:
+            raster_args["layer_name"] = "Raster"
+
+        if colormap_options is None:
+            colormap_options = [
+                "jet",
+                "viridis",
+                "plasma",
+                "inferno",
+                "magma",
+                "cividis",
+            ]
+
+        main_widget = widgets.VBox(layout=widgets.Layout(width="230px"))
+        style = {"description_width": "initial"}
+        layout = widgets.Layout(width="95%", padding="0px 5px 0px 5px")
+
+        interpolation_checkbox = widgets.Checkbox(
+            value=True,
+            description="Use interpolation",
+            style=style,
+            layout=layout,
+        )
+
+        threshold_slider = widgets.FloatSlider(
+            value=0.7,
+            min=0,
+            max=1,
+            step=0.01,
+            description="Threshold",
+            style=style,
+            layout=layout,
+        )
+
+        opacity_slider = widgets.FloatSlider(
+            value=0.5,
+            min=0,
+            max=1,
+            step=0.01,
+            description="Opacity",
+            style=style,
+            layout=layout,
+        )
+        colormap_dropdown = widgets.Dropdown(
+            options=colormap_options,
+            value="jet",
+            description="Colormap",
+            style=style,
+            layout=layout,
+        )
+        layer_name_input = widgets.Text(
+            value="Similarity",
+            description="Layer name",
+            style=style,
+            layout=layout,
+        )
+
+        save_button = widgets.Button(
+            description="Save",
+        )
+
+        reset_button = widgets.Button(
+            description="Reset",
+        )
+
+        output = widgets.Output()
+
+        main_widget.children = [
+            interpolation_checkbox,
+            threshold_slider,
+            opacity_slider,
+            colormap_dropdown,
+            layer_name_input,
+            widgets.HBox([save_button, reset_button]),
+            output,
+        ]
+
+        if host_map is not None:
+
+            host_map.add_widget(main_widget, add_header=True, position=position)
+
+            if raster is not None:
+                host_map.add_raster(raster, **raster_args)
+
+            def handle_map_interaction(**kwargs):
+                try:
+                    if kwargs.get("type") == "click":
+                        latlon = kwargs.get("coordinates")
+                        with output:
+                            output.clear_output()
+
+                            results = processor.compute_similarity(
+                                source=raster,
+                                features=features,
+                                query_coords=latlon[::-1],
+                                output_dir="dinov3_results",
+                                use_interpolation=interpolation_checkbox.value,
+                                coord_crs="EPSG:4326",
+                            )
+                            array = results["image_dict"]["image"]
+                            binary_array = array > threshold_slider.value
+                            image = dict_to_image(results["image_dict"])
+                            binary_image = dict_to_image(
+                                {
+                                    "image": binary_array,
+                                    "crs": results["image_dict"]["crs"],
+                                    "bounds": results["image_dict"]["bounds"],
+                                }
+                            )
+                            host_map.add_raster(
+                                image,
+                                colormap=colormap_dropdown.value,
+                                opacity=opacity_slider.value,
+                                layer_name=layer_name_input.value,
+                                zoom_to_layer=False,
+                                overwrite=True,
+                            )
+                            host_map.add_raster(
+                                binary_image,
+                                colormap="jet",
+                                nodata=0,
+                                opacity=opacity_slider.value,
+                                layer_name="Foreground",
+                                zoom_to_layer=False,
+                                overwrite=True,
+                                visible=False,
+                            )
+                except Exception as e:
+                    with output:
+                        print(e)
+
+            host_map.on_interaction(handle_map_interaction)
+            host_map.default_style = {"cursor": "crosshair"}

geoai/train.py

@@ -3163,11 +3163,17 @@ def semantic_segmentation(
 
     # Detect file format based on extension
     input_ext = os.path.splitext(input_path)[1].lower()
-    is_geotiff = input_ext in [".tif", ".tiff"]
+    is_geotiff = input_ext in [".tif", ".tiff", ".jp2", ".img"]
+    formats = {
+        ".tif": "GeoTIFF",
+        ".tiff": "GeoTIFF",
+        ".jp2": "JP2OpenJPEG",
+        ".img": "IMG",
+    }
 
     if not quiet:
         print(
-            f"Input file format: {'GeoTIFF' if is_geotiff else 'Regular image'} ({input_ext})"
+            f"Input file format: {formats[input_ext] if is_geotiff else 'Regular image'} ({input_ext})"
         )
 
     # Load model

geoai/utils.py

@@ -64,7 +64,7 @@ def view_raster(
     client_args: Optional[Dict] = {"cors_all": False},
     basemap: Optional[str] = "OpenStreetMap",
     basemap_args: Optional[Dict] = None,
-    backend: Optional[str] = "folium",
+    backend: Optional[str] = "ipyleaflet",
     **kwargs: Any,
 ) -> Any:
     """
@@ -87,6 +87,7 @@ def view_raster(
         client_args (Optional[Dict], optional): Additional arguments for the client. Defaults to {"cors_all": False}.
         basemap (Optional[str], optional): The basemap to use. Defaults to "OpenStreetMap".
         basemap_args (Optional[Dict], optional): Additional arguments for the basemap. Defaults to None.
+        backend (Optional[str], optional): The backend to use. Defaults to "ipyleaflet".
         **kwargs (Any): Additional keyword arguments.
 
     Returns:
@@ -396,12 +397,18 @@ def dict_to_rioxarray(data_dict: Dict) -> xr.DataArray:
     """
 
     from affine import Affine
+    from collections import namedtuple
+
+    BoundingBox = namedtuple("BoundingBox", ["minx", "maxx", "miny", "maxy"])
 
     # Extract components from the dictionary
     crs = data_dict["crs"]
     bounds = data_dict["bounds"]
     image_tensor = data_dict["image"]
 
+    if hasattr(bounds, "left"):
+        bounds = BoundingBox(bounds.left, bounds.right, bounds.bottom, bounds.top)
+
     # Convert tensor to numpy array if needed
     if hasattr(image_tensor, "numpy"):
         # For PyTorch tensors
@@ -2620,6 +2627,11 @@ def export_geotiff_tiles(
         create_overview (bool): Whether to create an overview image of all tiles
         skip_empty_tiles (bool): If True, skip tiles with no features
     """
+
+    import logging
+
+    logging.getLogger("rasterio").setLevel(logging.ERROR)
+
     # Create output directories
     os.makedirs(out_folder, exist_ok=True)
     image_dir = os.path.join(out_folder, "images")

{geoai_py-0.10.0.dist-info → geoai_py-0.11.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geoai-py
-Version: 0.10.0
+Version: 0.11.1
 Summary: A Python package for using Artificial Intelligence (AI) with geospatial data
 Author-email: Qiusheng Wu <giswqs@gmail.com>
 License: MIT License

geoai_py-0.11.1.dist-info/RECORD

@@ -0,0 +1,21 @@
+geoai/__init__.py,sha256=bf-BQ8_LOIWMsCC_0MMLRSBoeskwphJeLXAqyAHtuG8,3851
+geoai/change_detection.py,sha256=XkJjMEU1nD8uX3-nQy7NEmz8cukVeSaRxKJHlrv8xPM,59636
+geoai/classify.py,sha256=0DcComVR6vKU4qWtH2oHVeXc7ZTcV0mFvdXRtlNmolo,35637
+geoai/detectron2.py,sha256=dOOFM9M9-6PV8q2A4-mnIPrz7yTo-MpEvDiAW34nl0w,14610
+geoai/dinov3.py,sha256=pr7yFyw9yXGT95QkGO3Cmu__22rMyZJr3lHIPiLQ83U,40114
+geoai/download.py,sha256=B0EwpQFndJknOKmwRfEEnnCJhplOAwcLyNzFuA6FjZs,47633
+geoai/extract.py,sha256=595MBcSaFx-gQLIEv5g3oEM90QA5In4L59GPVgBOlQc,122092
+geoai/geoai.py,sha256=UK3HLsg3u0hJQfSQ29avWm6iKcKW35GmQ0f6jDp0ywg,10110
+geoai/hf.py,sha256=HbfJfpO6XnANKhmFOBvpwULiC65TeMlnLNtyQHHmlKA,17248
+geoai/map_widgets.py,sha256=8S0WCAeH8f1jswtBJHzV_lGaO92er8P58GxxotbKUng,5862
+geoai/sam.py,sha256=O6S-kGiFn7YEcFbfWFItZZQOhnsm6-GlunxQLY0daEs,34345
+geoai/segment.py,sha256=yBGTxA-ti8lBpk7WVaBOp6yP23HkaulKJQk88acrmZ0,43788
+geoai/segmentation.py,sha256=7yEzBSKCyHW1dNssoK0rdvhxi2IXsIQIFSga817KdI4,11535
+geoai/train.py,sha256=r9eioaBpc2eg6hckkGVI3aGhQZffKas_UVRj-AWruu8,136049
+geoai/utils.py,sha256=Vg8vNuIZ2PvmwPGAIZHN_1dHw4jC9ddjM_GmdkFN1KA,300899
+geoai_py-0.11.1.dist-info/licenses/LICENSE,sha256=vN2L5U7cZ6ZkOHFmc8WiGlsogWsZc5dllMeNxnKVOZg,1070
+geoai_py-0.11.1.dist-info/METADATA,sha256=7RYVHCX_BzOQXPiXg9hsejDQM4lUQaLKvO-22xLq0ow,6764
+geoai_py-0.11.1.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
+geoai_py-0.11.1.dist-info/entry_points.txt,sha256=uGp3Az3HURIsRHP9v-ys0hIbUuBBNUfXv6VbYHIXeg4,41
+geoai_py-0.11.1.dist-info/top_level.txt,sha256=1YkCUWu-ii-0qIex7kbwAvfei-gos9ycyDyUCJPNWHY,6
+geoai_py-0.11.1.dist-info/RECORD,,

geoai_py-0.10.0.dist-info/RECORD

@@ -1,19 +0,0 @@
-geoai/__init__.py,sha256=NaZb3vKclD1g50Cp45BvKY-dLnYC8ilz1J6wyRAgPY4,3766
-geoai/change_detection.py,sha256=XkJjMEU1nD8uX3-nQy7NEmz8cukVeSaRxKJHlrv8xPM,59636
-geoai/classify.py,sha256=0DcComVR6vKU4qWtH2oHVeXc7ZTcV0mFvdXRtlNmolo,35637
-geoai/detectron2.py,sha256=dOOFM9M9-6PV8q2A4-mnIPrz7yTo-MpEvDiAW34nl0w,14610
-geoai/download.py,sha256=B0EwpQFndJknOKmwRfEEnnCJhplOAwcLyNzFuA6FjZs,47633
-geoai/extract.py,sha256=595MBcSaFx-gQLIEv5g3oEM90QA5In4L59GPVgBOlQc,122092
-geoai/geoai.py,sha256=ZPr7hyJhOnwPO9c-nVJVaOUqMRZ77UpK95TFjKzDt0A,9782
-geoai/hf.py,sha256=HbfJfpO6XnANKhmFOBvpwULiC65TeMlnLNtyQHHmlKA,17248
-geoai/sam.py,sha256=O6S-kGiFn7YEcFbfWFItZZQOhnsm6-GlunxQLY0daEs,34345
-geoai/segment.py,sha256=yBGTxA-ti8lBpk7WVaBOp6yP23HkaulKJQk88acrmZ0,43788
-geoai/segmentation.py,sha256=7yEzBSKCyHW1dNssoK0rdvhxi2IXsIQIFSga817KdI4,11535
-geoai/train.py,sha256=tf29H_1GojczQnBzJpNPYXYfo1dDT7W1EivMlpXiEjM,135893
-geoai/utils.py,sha256=c171Gpk6WF5c2l4Hk9NYBGwSifGf-AuBKnA_SlnoTuA,300492
-geoai_py-0.10.0.dist-info/licenses/LICENSE,sha256=vN2L5U7cZ6ZkOHFmc8WiGlsogWsZc5dllMeNxnKVOZg,1070
-geoai_py-0.10.0.dist-info/METADATA,sha256=fwxfmIApQl0bmwZmVj-A8X0vvPfELJOCp0QllFfNxyE,6764
-geoai_py-0.10.0.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
-geoai_py-0.10.0.dist-info/entry_points.txt,sha256=uGp3Az3HURIsRHP9v-ys0hIbUuBBNUfXv6VbYHIXeg4,41
-geoai_py-0.10.0.dist-info/top_level.txt,sha256=1YkCUWu-ii-0qIex7kbwAvfei-gos9ycyDyUCJPNWHY,6
-geoai_py-0.10.0.dist-info/RECORD,,