geoai-py 0.4.3__tar.gz → 0.5.0__tar.gz

{geoai_py-0.4.3 → geoai_py-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geoai-py
-Version: 0.4.3
+Version: 0.5.0
 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.5.0/docs/classify.md

@@ -0,0 +1,3 @@
+# classify module
+
+::: geoai.classify

{geoai_py-0.4.3 → geoai_py-0.5.0}/docs/examples/building_footprints_africa.ipynb

@@ -76,9 +76,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "extractor = geoai.BuildingFootprintExtractor(\n",
-    "    model_path=\"building_footprints_africa.pth\"\n",
-    ")"
+    "extractor = geoai.BuildingFootprintExtractor(model_path=\"building_footprints_usa.pth\")"
    ]
   },
   {

{geoai_py-0.4.3 → geoai_py-0.5.0}/docs/examples/building_footprints_china.ipynb

@@ -77,7 +77,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "extractor = geoai.BuildingFootprintExtractor(model_path=\"building_footprints_china.pth\")"
+    "extractor = geoai.BuildingFootprintExtractor(model_path=\"building_footprints_usa.pth\")"
    ]
   },
   {

{geoai_py-0.4.3 → geoai_py-0.5.0}/docs/examples/download_data.ipynb

@@ -122,10 +122,7 @@
     "# Download buildings\n",
     "data_file = download_overture_buildings(\n",
     "    bbox=bbox,\n",
-    "    output_file=\"buildings.geojson\",\n",
-    "    output_format=\"geojson\",\n",
-    "    data_type=\"building\",\n",
-    "    verbose=True,\n",
+    "    output=\"buildings.geojson\",\n",
     ")"
    ]
   },
@@ -143,9 +140,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "if data_file:\n",
-    "    stats = extract_building_stats(data_file)\n",
-    "    print(stats)"
+    "stats = extract_building_stats(data_file)\n",
+    "print(stats)"
    ]
   },
   {
@@ -177,7 +173,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "geo",
    "language": "python",
    "name": "python3"
   },
@@ -191,7 +187,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.8"
+   "version": "3.12.9"
   }
  },
  "nbformat": 4,

geoai_py-0.5.0/docs/examples/train_water_detection.ipynb

@@ -0,0 +1,328 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Train a Model for Detecting Surface Water\n",
+    "\n",
+    "[![image](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/opengeos/geoai/blob/main/docs/examples/train_water_detection.ipynb)\n",
+    "\n",
+    "## Install package\n",
+    "To use the `geoai-py` package, ensure it is installed in your environment. Uncomment the command below if needed."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# %pip install geoai-py"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Import libraries"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import geoai"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Download sample data"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "train_raster_url = \"https://huggingface.co/datasets/giswqs/geospatial/resolve/main/naip/naip_water_train.tif\"\n",
+    "train_masks_url = \"https://huggingface.co/datasets/giswqs/geospatial/resolve/main/naip/naip_water_masks.tif\"\n",
+    "test_raster_url = \"https://huggingface.co/datasets/giswqs/geospatial/resolve/main/naip/naip_water_test.tif\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "train_raster_path = geoai.download_file(train_raster_url)\n",
+    "train_masks_path = geoai.download_file(train_masks_url)\n",
+    "test_raster_path = geoai.download_file(test_raster_url)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "geoai.print_raster_info(train_raster_path, show_preview=False)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Visualize sample data"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "geoai.view_raster(train_masks_url, nodata=0, basemap=train_raster_url)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "geoai.view_raster(test_raster_url)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Create training data"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "out_folder = \"output\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "tiles = geoai.export_geotiff_tiles(\n",
+    "    in_raster=train_raster_path,\n",
+    "    out_folder=out_folder,\n",
+    "    in_class_data=train_masks_path,\n",
+    "    tile_size=512,\n",
+    "    stride=128,\n",
+    "    buffer_radius=0,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Train object detection model"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "geoai.train_MaskRCNN_model(\n",
+    "    images_dir=f\"{out_folder}/images\",\n",
+    "    labels_dir=f\"{out_folder}/labels\",\n",
+    "    output_dir=f\"{out_folder}/models\",\n",
+    "    num_channels=4,\n",
+    "    pretrained=True,\n",
+    "    batch_size=4,\n",
+    "    num_epochs=10,\n",
+    "    learning_rate=0.005,\n",
+    "    val_split=0.2,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Run inference"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "masks_path = \"naip_water_prediction.tif\"\n",
+    "model_path = f\"{out_folder}/models/best_model.pth\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "geoai.object_detection(\n",
+    "    test_raster_path,\n",
+    "    masks_path,\n",
+    "    model_path,\n",
+    "    window_size=512,\n",
+    "    overlap=128,\n",
+    "    confidence_threshold=0.3,\n",
+    "    batch_size=4,\n",
+    "    num_channels=4,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Vectorize masks"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "output_path = \"naip_water_prediction.geojson\"\n",
+    "gdf = geoai.raster_to_vector(\n",
+    "    masks_path, output_path, min_area=1000, simplify_tolerance=1\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "gdf = geoai.add_geometric_properties(gdf)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "len(gdf)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "geoai.view_vector_interactive(gdf, tiles=test_raster_url)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "gdf[\"elongation\"].hist()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "gdf_filtered = gdf[gdf[\"elongation\"] < 10]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "len(gdf_filtered)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Visualize results"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "geoai.view_vector_interactive(gdf_filtered, tiles=test_raster_url)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "geoai.create_split_map(\n",
+    "    left_layer=gdf_filtered,\n",
+    "    right_layer=test_raster_url,\n",
+    "    left_args={\"style\": {\"color\": \"red\", \"fillOpacity\": 0.2}},\n",
+    "    basemap=test_raster_url,\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "![image](https://github.com/user-attachments/assets/a269b5a0-9f72-4ed8-8b2d-a175bbc45a23)"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.12.9"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 4
+}

{geoai_py-0.4.3 → geoai_py-0.5.0}/geoai/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = """Qiusheng Wu"""
 __email__ = "giswqs@gmail.com"
-__version__ = "0.4.3"
+__version__ = "0.5.0"
 
 
 import os
@@ -23,6 +23,8 @@ def set_proj_lib_path(verbose=False):
         bool: True if both paths were set successfully, False otherwise.
     """
     try:
+        from rasterio.env import set_gdal_config
+
         # Get conda environment path
         conda_env_path = os.environ.get("CONDA_PREFIX") or sys.prefix
 
@@ -86,6 +88,9 @@ def set_proj_lib_path(verbose=False):
                 if proj_set and gdal_set:
                     break
 
+        set_gdal_config("PROJ_LIB", os.environ["PROJ_LIB"])
+        set_gdal_config("GDAL_DATA", os.environ["GDAL_DATA"])
+
     except Exception as e:
         print(f"Error setting projection library paths: {e}")
         return