geoai-py 0.3.6__py2.py3-none-any.whl → 0.4.1__py2.py3-none-any.whl

geoai/utils.py

@@ -28,25 +28,11 @@ from rasterio import features
 from rasterio.plot import show
 from rasterio.windows import Window
 from shapely.affinity import rotate
-from shapely.geometry import (
-    MultiPolygon,
-    Polygon,
-    box,
-    mapping,
-    shape,
-)
+from shapely.geometry import MultiPolygon, Polygon, box, mapping, shape
 from torchvision.transforms import RandomRotation
 from tqdm import tqdm
 
 
-try:
-    from torchgeo.datasets import RasterDataset, unbind_samples
-except ImportError as e:
-    raise ImportError(
-        "Your torchgeo version is too old. Please upgrade to the latest version using 'pip install -U torchgeo'."
-    )
-
-
 def view_raster(
     source: str,
     indexes: Optional[int] = None,
@@ -60,7 +46,7 @@ def view_raster(
     zoom_to_layer: Optional[bool] = True,
     visible: Optional[bool] = True,
     opacity: Optional[float] = 1.0,
-    array_args: Optional[Dict] = {},
+    array_args: Optional[Dict] = None,
     client_args: Optional[Dict] = {"cors_all": False},
     basemap: Optional[str] = "OpenStreetMap",
     basemap_args: Optional[Dict] = None,
@@ -101,6 +87,9 @@ def view_raster(
     if basemap_args is None:
         basemap_args = {}
 
+    if array_args is None:
+        array_args = {}
+
     m = leafmap.Map()
 
     if isinstance(basemap, str):
@@ -284,6 +273,14 @@ def plot_batch(
     Returns:
         None
     """
+
+    try:
+        from torchgeo.datasets import unbind_samples
+    except ImportError as e:
+        raise ImportError(
+            "Your torchgeo version is too old. Please upgrade to the latest version using 'pip install -U torchgeo'."
+        )
+
     # Get the samples and the number of items in the batch
     samples = unbind_samples(batch.copy())
 
@@ -323,9 +320,7 @@ def plot_batch(
             )
 
 
-def calc_stats(
-    dataset: RasterDataset, divide_by: float = 1.0
-) -> Tuple[np.ndarray, np.ndarray]:
+def calc_stats(dataset, divide_by: float = 1.0) -> Tuple[np.ndarray, np.ndarray]:
     """
     Calculate the statistics (mean and std) for the entire dataset.
 
@@ -700,8 +695,8 @@ def view_vector_interactive(
     """
     import folium
     import folium.plugins as plugins
-    from localtileserver import get_folium_tile_layer, TileClient
     from leafmap import cog_tile
+    from localtileserver import TileClient, get_folium_tile_layer
 
     google_tiles = {
         "Roadmap": {
@@ -803,9 +798,9 @@ def regularization(
     Returns:
         GeoDataFrame or list of shapely Polygons with regularized building footprints
     """
-    from shapely.geometry import Polygon, shape
-    from shapely.affinity import rotate, translate
     from shapely import wkt
+    from shapely.affinity import rotate, translate
+    from shapely.geometry import Polygon, shape
 
     regularized_buildings = []
 
@@ -924,8 +919,8 @@ def hybrid_regularization(building_polygons):
     Returns:
         GeoDataFrame or list of shapely Polygons with regularized building footprints
     """
-    from shapely.geometry import Polygon
     from shapely.affinity import rotate
+    from shapely.geometry import Polygon
 
     # Use minimum_rotated_rectangle instead of oriented_envelope
     try:
@@ -1039,8 +1034,8 @@ def adaptive_regularization(
     Returns:
         GeoDataFrame or list of shapely Polygons with regularized building footprints
     """
-    from shapely.geometry import Polygon
     from shapely.affinity import rotate
+    from shapely.geometry import Polygon
 
     # Analyze the overall dataset to set appropriate parameters
     if is_gdf := isinstance(building_polygons, gpd.GeoDataFrame):
@@ -2594,6 +2589,8 @@ def export_geotiff_tiles(
             print(f"\nRaster info for {in_raster}:")
             print(f"  CRS: {src.crs}")
             print(f"  Dimensions: {src.width} x {src.height}")
+            print(f"  Resolution: {src.res}")
+            print(f"  Bands: {src.count}")
             print(f"  Bounds: {src.bounds}")
 
         # Calculate number of tiles
@@ -4275,9 +4272,10 @@ def read_vector(source, layer=None, **kwargs):
         >>> gdf = read_vector("path/to/data.gpkg", layer="layer_name")
     """
 
-    import fiona
     import urllib.parse
 
+    import fiona
+
     # Determine if source is a URL or local file
     parsed_url = urllib.parse.urlparse(source)
     is_url = parsed_url.scheme in ["http", "https"]
@@ -4351,6 +4349,7 @@ def read_raster(source, band=None, masked=True, **kwargs):
         >>> raster = read_raster("path/to/data.tif", masked=False)
     """
     import urllib.parse
+
     from rasterio.errors import RasterioIOError
 
     # Determine if source is a URL or local file
@@ -4445,8 +4444,8 @@ def region_groups(
     Returns:
         Union[Tuple[np.ndarray, pd.DataFrame], Tuple[xr.DataArray, pd.DataFrame]]: Labeled image and properties DataFrame.
     """
-    from skimage import measure
     import scipy.ndimage as ndi
+    from skimage import measure
 
     if isinstance(image, str):
         ds = rxr.open_rasterio(image)
@@ -5670,11 +5669,14 @@ def orthogonalize(
         crs = src.crs
 
         # Extract shapes from the raster mask
-        shapes = features.shapes(mask, transform=transform)
+        shapes = list(features.shapes(mask, transform=transform))
+
+        # Initialize progress bar
+        print(f"Processing {len(shapes)} features...")
 
         # Convert shapes to GeoJSON features
         features_list = []
-        for shape, value in shapes:
+        for shape, value in tqdm(shapes, desc="Converting features", unit="shape"):
             if value > 0:  # Only process non-zero values (actual objects)
                 # Convert GeoJSON geometry to Shapely polygon
                 polygon = Polygon(shape["coordinates"][0])
@@ -5814,6 +5816,8 @@ def orthogonalize(
 
         # Save to file if output_path is provided
         if output_path:
+            print(f"Saving to {output_path}...")
             gdf.to_file(output_path)
+            print("Done!")
 
         return gdf

{geoai_py-0.3.6.dist-info → geoai_py-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: geoai-py
-Version: 0.3.6
+Version: 0.4.1
 Summary: A Python package for using Artificial Intelligence (AI) with geospatial data
 Author-email: Qiusheng Wu <giswqs@gmail.com>
 License: MIT License
@@ -36,13 +36,8 @@ Requires-Dist: torch
 Requires-Dist: torchgeo
 Requires-Dist: tqdm
 Requires-Dist: transformers
-Provides-Extra: download
-Requires-Dist: pystac_client; extra == "download"
-Requires-Dist: planetary_computer; extra == "download"
-Requires-Dist: tqdm; extra == "download"
-Requires-Dist: overturemaps; extra == "download"
-Provides-Extra: all
-Requires-Dist: geoai[download]; extra == "all"
+Provides-Extra: extra
+Requires-Dist: overturemaps; extra == "extra"
 
 # GeoAI: Artificial Intelligence for Geospatial Data
 

geoai_py-0.4.1.dist-info/RECORD

@@ -0,0 +1,15 @@
+geoai/__init__.py,sha256=lZ5LYzlwjX-TuvVtvuk0TC0le80P83lfUUOXBY0MPoU,3592
+geoai/download.py,sha256=eqMecJqvqyrIVFViNA7pW8a5EIhqYJzRILmxQoFHG2k,13095
+geoai/extract.py,sha256=CCXjUcGC4ZOKOKKjvElp8VFmTz46b0ATvGitbOPgTwE,95506
+geoai/geoai.py,sha256=L1jkozDcjqJXvqT6i8oW04Ix9x4cc2-LNYi9_564ABQ,163
+geoai/hf.py,sha256=mLKGxEAS5eHkxZLwuLpYc1o7e3-7QIXdBv-QUY-RkFk,17072
+geoai/segment.py,sha256=g3YW17ftr--CKq6VB32TJEPY8owGQ7uQ0sg_tUT2ooE,13681
+geoai/segmentation.py,sha256=AtPzCvguHAEeuyXafa4bzMFATvltEYcah1B8ZMfkM_s,11373
+geoai/train.py,sha256=VaeFzIkVUNTdre8ImgUNhmbpA42qijSXaajLpmBF_Ic,36248
+geoai/utils.py,sha256=oBNhk73_Owv-pmaRyBKkq0HinnqnMgP3U5CUAQx6ln0,223700
+geoai_py-0.4.1.dist-info/LICENSE,sha256=vN2L5U7cZ6ZkOHFmc8WiGlsogWsZc5dllMeNxnKVOZg,1070
+geoai_py-0.4.1.dist-info/METADATA,sha256=0JZwrVh1EtR3mJIu8cumnafbCnElW70iBa77hSDwXHk,6078
+geoai_py-0.4.1.dist-info/WHEEL,sha256=SrDKpSbFN1G94qcmBqS9nyHcDMp9cUS9OC06hC0G3G0,109
+geoai_py-0.4.1.dist-info/entry_points.txt,sha256=uGp3Az3HURIsRHP9v-ys0hIbUuBBNUfXv6VbYHIXeg4,41
+geoai_py-0.4.1.dist-info/top_level.txt,sha256=1YkCUWu-ii-0qIex7kbwAvfei-gos9ycyDyUCJPNWHY,6
+geoai_py-0.4.1.dist-info/RECORD,,

geoai_py-0.3.6.dist-info/RECORD

@@ -1,13 +0,0 @@
-geoai/__init__.py,sha256=vsoVgC5QbrVHI4xAGVZLh7C1RxBLKFdGk8KubSwKzTE,923
-geoai/download.py,sha256=4GiDmLrp2wKslgfm507WeZrwOdYcMekgQXxWGbl5cBw,13094
-geoai/extract.py,sha256=v_4i_pSQYrdpNZZsGvVaa6Lo0AC7bHUaNp5bs_O4kS4,94255
-geoai/geoai.py,sha256=di-lh7704BpwltR5cxz09AtCx2EaNkvtUJxRakkYPpc,87
-geoai/segment.py,sha256=ULe0Xq2RnA7p6d8vKm0rYqHQW6G_3dOxpd_1W0x9UOU,13680
-geoai/segmentation.py,sha256=Vcymnhwl_xikt4v9x8CYJq_vId9R1gB7-YzLfwg-F9M,11372
-geoai/utils.py,sha256=DKHLLLus-R12pJCFknPvktXbju8nC8SSYcsbDjykovg,223372
-geoai_py-0.3.6.dist-info/LICENSE,sha256=vN2L5U7cZ6ZkOHFmc8WiGlsogWsZc5dllMeNxnKVOZg,1070
-geoai_py-0.3.6.dist-info/METADATA,sha256=wzNDh3GiOtZVzA4QDhuUWc3KTtNVqwvc4rDQSgyfeFc,6297
-geoai_py-0.3.6.dist-info/WHEEL,sha256=SrDKpSbFN1G94qcmBqS9nyHcDMp9cUS9OC06hC0G3G0,109
-geoai_py-0.3.6.dist-info/entry_points.txt,sha256=uGp3Az3HURIsRHP9v-ys0hIbUuBBNUfXv6VbYHIXeg4,41
-geoai_py-0.3.6.dist-info/top_level.txt,sha256=1YkCUWu-ii-0qIex7kbwAvfei-gos9ycyDyUCJPNWHY,6
-geoai_py-0.3.6.dist-info/RECORD,,