geoai-py 0.4.3__py2.py3-none-any.whl → 0.5.1__py2.py3-none-any.whl

geoai/utils.py

@@ -6053,3 +6053,199 @@ def try_common_architectures(state_dict):
 
         except Exception as e:
             print(f"- {name}: Failed to load - {str(e)}")
+
+
+def mosaic_geotiffs(input_dir, output_file, mask_file=None):
+    """Create a mosaic from all GeoTIFF files as a Cloud Optimized GeoTIFF (COG).
+
+    This function identifies all GeoTIFF files in the specified directory,
+    creates a seamless mosaic with proper handling of nodata values, and saves
+    as a Cloud Optimized GeoTIFF format. If a mask file is provided, the output
+    will be clipped to the extent of the mask.
+
+    Args:
+        input_dir (str): Path to the directory containing GeoTIFF files.
+        output_file (str): Path to the output Cloud Optimized GeoTIFF file.
+        mask_file (str, optional): Path to a mask file to clip the output.
+            If provided, the output will be clipped to the extent of this mask.
+            Defaults to None.
+
+    Returns:
+        bool: True if the mosaic was created successfully, False otherwise.
+
+    Examples:
+        >>> mosaic_geotiffs('naip', 'merged_naip.tif')
+        True
+        >>> mosaic_geotiffs('naip', 'merged_naip.tif', 'boundary.tif')
+        True
+    """
+    import glob
+    from osgeo import gdal
+
+    gdal.UseExceptions()
+    # Get all tif files in the directory
+    tif_files = glob.glob(os.path.join(input_dir, "*.tif"))
+
+    if not tif_files:
+        print("No GeoTIFF files found in the specified directory.")
+        return False
+
+    # Analyze the first input file to determine compression and nodata settings
+    ds = gdal.Open(tif_files[0])
+    if ds is None:
+        print(f"Unable to open {tif_files[0]}")
+        return False
+
+    # Get driver metadata from the first file
+    driver = ds.GetDriver()
+    creation_options = []
+
+    # Check compression type
+    metadata = ds.GetMetadata("IMAGE_STRUCTURE")
+    if "COMPRESSION" in metadata:
+        compression = metadata["COMPRESSION"]
+        creation_options.append(f"COMPRESS={compression}")
+    else:
+        # Default compression if none detected
+        creation_options.append("COMPRESS=LZW")
+
+    # Add COG-specific creation options
+    creation_options.extend(["TILED=YES", "BLOCKXSIZE=512", "BLOCKYSIZE=512"])
+
+    # Check for nodata value in the first band of the first file
+    band = ds.GetRasterBand(1)
+    has_nodata = band.GetNoDataValue() is not None
+    nodata_value = band.GetNoDataValue() if has_nodata else None
+
+    # Close the dataset
+    ds = None
+
+    # Create a temporary VRT (Virtual Dataset)
+    vrt_path = os.path.join(input_dir, "temp_mosaic.vrt")
+
+    # Build VRT from input files with proper nodata handling
+    vrt_options = gdal.BuildVRTOptions(
+        resampleAlg="nearest",
+        srcNodata=nodata_value if has_nodata else None,
+        VRTNodata=nodata_value if has_nodata else None,
+    )
+    vrt_dataset = gdal.BuildVRT(vrt_path, tif_files, options=vrt_options)
+
+    # Close the VRT dataset to flush it to disk
+    vrt_dataset = None
+
+    # Create temp mosaic
+    temp_mosaic = output_file + ".temp.tif"
+
+    # Convert VRT to GeoTIFF with the same compression as input
+    translate_options = gdal.TranslateOptions(
+        format="GTiff",
+        creationOptions=creation_options,
+        noData=nodata_value if has_nodata else None,
+    )
+    gdal.Translate(temp_mosaic, vrt_path, options=translate_options)
+
+    # Apply mask if provided
+    if mask_file and os.path.exists(mask_file):
+        print(f"Clipping mosaic to mask: {mask_file}")
+
+        # Create a temporary clipped file
+        clipped_mosaic = output_file + ".clipped.tif"
+
+        # Open mask file
+        mask_ds = gdal.Open(mask_file)
+        if mask_ds is None:
+            print(f"Unable to open mask file: {mask_file}")
+            # Continue without clipping
+        else:
+            # Get mask extent
+            mask_geotransform = mask_ds.GetGeoTransform()
+            mask_projection = mask_ds.GetProjection()
+            mask_ulx = mask_geotransform[0]
+            mask_uly = mask_geotransform[3]
+            mask_lrx = mask_ulx + (mask_geotransform[1] * mask_ds.RasterXSize)
+            mask_lry = mask_uly + (mask_geotransform[5] * mask_ds.RasterYSize)
+
+            # Close mask dataset
+            mask_ds = None
+
+            # Use warp options to clip
+            warp_options = gdal.WarpOptions(
+                format="GTiff",
+                outputBounds=[mask_ulx, mask_lry, mask_lrx, mask_uly],
+                dstSRS=mask_projection,
+                creationOptions=creation_options,
+                srcNodata=nodata_value if has_nodata else None,
+                dstNodata=nodata_value if has_nodata else None,
+            )
+
+            # Apply clipping
+            gdal.Warp(clipped_mosaic, temp_mosaic, options=warp_options)
+
+            # Remove the unclipped temp mosaic and use the clipped one
+            os.remove(temp_mosaic)
+            temp_mosaic = clipped_mosaic
+
+    # Create internal overviews for the temp mosaic
+    ds = gdal.Open(temp_mosaic, gdal.GA_Update)
+    overview_list = [2, 4, 8, 16, 32]
+    ds.BuildOverviews("NEAREST", overview_list)
+    ds = None  # Close the dataset to ensure overviews are written
+
+    # Convert the temp mosaic to a proper COG
+    cog_options = gdal.TranslateOptions(
+        format="GTiff",
+        creationOptions=[
+            "TILED=YES",
+            "COPY_SRC_OVERVIEWS=YES",
+            "COMPRESS=DEFLATE",
+            "PREDICTOR=2",
+            "BLOCKXSIZE=512",
+            "BLOCKYSIZE=512",
+        ],
+        noData=nodata_value if has_nodata else None,
+    )
+    gdal.Translate(output_file, temp_mosaic, options=cog_options)
+
+    # Clean up temporary files
+    if os.path.exists(vrt_path):
+        os.remove(vrt_path)
+    if os.path.exists(temp_mosaic):
+        os.remove(temp_mosaic)
+
+    print(f"Cloud Optimized GeoTIFF mosaic created successfully: {output_file}")
+    return True
+
+
+def download_model_from_hf(model_path, repo_id=None):
+    """
+    Download the object detection model from Hugging Face.
+
+    Args:
+        model_path: Path to the model file.
+        repo_id: Hugging Face repository ID.
+
+    Returns:
+        Path to the downloaded model file
+    """
+    from huggingface_hub import hf_hub_download
+
+    try:
+
+        # Define the repository ID and model filename
+        if repo_id is None:
+            print(
+                "Repo is is not specified, using default Hugging Face repo_id: giswqs/geoai"
+            )
+            repo_id = "giswqs/geoai"
+
+        # Download the model
+        model_path = hf_hub_download(repo_id=repo_id, filename=model_path)
+        print(f"Model downloaded to: {model_path}")
+
+        return model_path
+
+    except Exception as e:
+        print(f"Error downloading model from Hugging Face: {e}")
+        print("Please specify a local model path or ensure internet connectivity.")
+        raise

{geoai_py-0.4.3.dist-info → geoai_py-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geoai-py
-Version: 0.4.3
+Version: 0.5.1
 Summary: A Python package for using Artificial Intelligence (AI) with geospatial data
 Author-email: Qiusheng Wu <giswqs@gmail.com>
 License: MIT License
@@ -25,6 +25,7 @@ Requires-Dist: jupyter-server-proxy
 Requires-Dist: leafmap
 Requires-Dist: localtileserver
 Requires-Dist: mapclassify
+Requires-Dist: maplibre
 Requires-Dist: overturemaps
 Requires-Dist: planetary-computer
 Requires-Dist: pystac-client
@@ -143,3 +144,10 @@ We welcome contributions of all kinds! See our [contributing guide](https://geoa
 ## 📄 License
 
 GeoAI is free and open source software, licensed under the MIT License.
+
+## Acknowledgments
+
+We gratefully acknowledge the support of the following organizations:
+
+-   [NASA](https://www.nasa.gov): This research is partially supported by the National Aeronautics and Space Administration (NASA) through Grant No. 80NSSC22K1742, awarded under the [Open Source Tools, Frameworks, and Libraries Program](https://bit.ly/3RVBRcQ).
+-   [AmericaView](https://americaview.org): This work is also partially supported by the U.S. Geological Survey through Grant/Cooperative Agreement No. G23AP00683 (GY23-GY27) in collaboration with AmericaView.

geoai_py-0.5.1.dist-info/RECORD

@@ -0,0 +1,16 @@
+geoai/__init__.py,sha256=pJWIhWx8tlrmE1KCdmmSNJFEyvGyzOBBtYWdtQanGCU,3765
+geoai/classify.py,sha256=_e-193QzAx3pIxUflPIsIs1qZevQx5ADu7i3bOL1G70,35055
+geoai/download.py,sha256=lJ1GsJOZsKc2i6_dQyPV-XXIXmlADOpmSBo-wha4DEU,40892
+geoai/extract.py,sha256=GocJufMmrwEWxNBL1J91EXXHL8AKcO8m_lmtUF5AKPw,119102
+geoai/geoai.py,sha256=QGKDAenLPo1CgKZLsMFDzW7CLH1Ao2sYaWhFZG3eTpk,4896
+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=mQXat2yuddT-2rME4xnX_m3SkY23E_-zdxLnBIKxw8o,44091
+geoai/utils.py,sha256=5BZTL9QlJGEs9uw5w6i_aZ4s8SH_FGvb6ZFlIyEHEZI,239703
+geoai_py-0.5.1.dist-info/licenses/LICENSE,sha256=vN2L5U7cZ6ZkOHFmc8WiGlsogWsZc5dllMeNxnKVOZg,1070
+geoai_py-0.5.1.dist-info/METADATA,sha256=9_A_dN9Vnl5C4Q_i7N8OdgTTo1Ke4wunve3ChsUvQnA,6637
+geoai_py-0.5.1.dist-info/WHEEL,sha256=MAQBAzGbXNI3bUmkDsiV_duv8i-gcdnLzw7cfUFwqhU,109
+geoai_py-0.5.1.dist-info/entry_points.txt,sha256=uGp3Az3HURIsRHP9v-ys0hIbUuBBNUfXv6VbYHIXeg4,41
+geoai_py-0.5.1.dist-info/top_level.txt,sha256=1YkCUWu-ii-0qIex7kbwAvfei-gos9ycyDyUCJPNWHY,6
+geoai_py-0.5.1.dist-info/RECORD,,

{geoai_py-0.4.3.dist-info → geoai_py-0.5.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (77.0.3)
+Generator: setuptools (78.1.0)
 Root-Is-Purelib: true
 Tag: py2-none-any
 Tag: py3-none-any

geoai_py-0.4.3.dist-info/RECORD

@@ -1,15 +0,0 @@
-geoai/__init__.py,sha256=b12E2HztHEPaaKMVKpk6GPjC7ElUMxOW9pYKl4VZkmE,3592
-geoai/download.py,sha256=BvCEpcBwZlEOWieixsdyvQDiE0CXRjU7oayLmy5_Dgs,40110
-geoai/extract.py,sha256=GocJufMmrwEWxNBL1J91EXXHL8AKcO8m_lmtUF5AKPw,119102
-geoai/geoai.py,sha256=BqKdWzNruDdGqwqoyTaJzUq4lKGj-RDBZlSO3t3-GxQ,626
-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=-l2j1leTxDnFDLaBslu1q6CobXjm3LEdiQwUWOU8P6M,40088
-geoai/utils.py,sha256=Wg9jbMBKUZSGUmU8Vkp6v19QcDNg5KmcyZxuHqJvgnc,233016
-geoai_py-0.4.3.dist-info/licenses/LICENSE,sha256=vN2L5U7cZ6ZkOHFmc8WiGlsogWsZc5dllMeNxnKVOZg,1070
-geoai_py-0.4.3.dist-info/METADATA,sha256=geDmJ-1zHImsOdcj4gypgq8JqSy8MznnxAnICwh0EbA,6049
-geoai_py-0.4.3.dist-info/WHEEL,sha256=aoLN90hLOL0c0qxXMxWYUM3HA3WmFGZQqEJHX1V_OJE,109
-geoai_py-0.4.3.dist-info/entry_points.txt,sha256=uGp3Az3HURIsRHP9v-ys0hIbUuBBNUfXv6VbYHIXeg4,41
-geoai_py-0.4.3.dist-info/top_level.txt,sha256=1YkCUWu-ii-0qIex7kbwAvfei-gos9ycyDyUCJPNWHY,6
-geoai_py-0.4.3.dist-info/RECORD,,