geoai-py 0.5.6__tar.gz → 0.6.0__tar.gz

geoai_py-0.6.0/.github/workflows/docker-image.yml

@@ -0,0 +1,16 @@
+name: Docker Image CI
+
+on:
+    push:
+        branches: ["main"]
+    pull_request:
+        branches: ["main"]
+
+jobs:
+    build:
+        runs-on: ubuntu-latest
+
+        steps:
+            - uses: actions/checkout@v4
+            - name: Build the Docker image
+              run: docker build . --file Dockerfile --tag ${{ github.repository }}:$(date +%s)

geoai_py-0.6.0/.github/workflows/docker-publish.yml

@@ -0,0 +1,45 @@
+name: Publish Docker image
+
+on:
+    release:
+        types: [published]
+
+jobs:
+    push_to_registries:
+        name: Push Docker image to multiple registries
+        runs-on: ubuntu-latest
+        permissions:
+            packages: write
+            contents: read
+        steps:
+            - name: Check out the repo
+              uses: actions/checkout@v4
+
+            - name: Log in to Docker Hub
+              uses: docker/login-action@v3
+              with:
+                  username: ${{ secrets.DOCKER_USERNAME }}
+                  password: ${{ secrets.DOCKER_PASSWORD }}
+
+            - name: Log in to the Container registry
+              uses: docker/login-action@v3
+              with:
+                  registry: ghcr.io
+                  username: ${{ github.actor }}
+                  password: ${{ secrets.GITHUB_TOKEN }}
+
+            - name: Extract metadata (tags, labels) for Docker
+              id: meta
+              uses: docker/metadata-action@v5
+              with:
+                  images: |
+                      giswqs/geoai
+                      ghcr.io/${{ github.repository }}
+
+            - name: Build and push Docker images
+              uses: docker/build-push-action@v6
+              with:
+                  context: .
+                  push: true
+                  tags: ${{ steps.meta.outputs.tags }}
+                  labels: ${{ steps.meta.outputs.labels }}

{geoai_py-0.5.6 → geoai_py-0.6.0}/.github/workflows/docs-build.yml

@@ -17,7 +17,7 @@ jobs:
                   fetch-depth: 0
 
             - name: Install uv
-              uses: astral-sh/setup-uv@v5
+              uses: astral-sh/setup-uv@v6
               with:
                   version: "0.4.16"
                   enable-cache: false

{geoai_py-0.5.6 → geoai_py-0.6.0}/.github/workflows/docs.yml

@@ -16,7 +16,7 @@ jobs:
                   fetch-depth: 0
 
             - name: Install uv
-              uses: astral-sh/setup-uv@v5
+              uses: astral-sh/setup-uv@v6
               with:
                   version: "0.4.16"
                   enable-cache: false

{geoai_py-0.5.6 → geoai_py-0.6.0}/.github/workflows/macos.yml

@@ -22,7 +22,7 @@ jobs:
             - uses: actions/checkout@v4
 
             - name: Install uv
-              uses: astral-sh/setup-uv@v5
+              uses: astral-sh/setup-uv@v6
               with:
                   version: "0.4.16"
                   enable-cache: false

{geoai_py-0.5.6 → geoai_py-0.6.0}/.github/workflows/ubuntu.yml

@@ -20,7 +20,7 @@ jobs:
             - uses: actions/checkout@v4
 
             - name: Install uv
-              uses: astral-sh/setup-uv@v5
+              uses: astral-sh/setup-uv@v6
               with:
                   version: "0.4.16"
                   enable-cache: false

{geoai_py-0.5.6 → geoai_py-0.6.0}/.github/workflows/windows.yml

@@ -18,7 +18,7 @@ jobs:
             - uses: actions/checkout@v4
 
             - name: Install uv
-              uses: astral-sh/setup-uv@v5
+              uses: astral-sh/setup-uv@v6
               with:
                   version: "0.4.16"
                   enable-cache: false

geoai_py-0.6.0/Dockerfile

@@ -0,0 +1,54 @@
+FROM quay.io/jupyter/base-notebook:latest
+
+# -------------------------------------------------------
+# 1. Install system-level packages (minimal, just git)
+# -------------------------------------------------------
+USER root
+RUN apt-get update && \
+    apt-get install -y git && \
+    rm -rf /var/lib/apt/lists/*
+
+# -------------------------------------------------------
+# 2. Install geospatial Python packages via conda (base env)
+# -------------------------------------------------------
+RUN mamba install -n base -c conda-forge \
+    geoai \
+    overturemaps -y && \
+    fix-permissions "${CONDA_DIR}"
+
+# -------------------------------------------------------
+# 3. Environment variables
+# -------------------------------------------------------
+ENV PROJ_LIB=/opt/conda/share/proj
+ENV GDAL_DATA=/opt/conda/share/gdal
+ENV LOCALTILESERVER_CLIENT_PREFIX='proxy/{port}'
+
+# -------------------------------------------------------
+# 4. Copy source code (do this *after* package installs to improve caching)
+# -------------------------------------------------------
+COPY . /home/jovyan/geoai
+WORKDIR /home/jovyan/geoai
+
+
+# -------------------------------------------------------
+# 5. Build and install geoai from source
+# -------------------------------------------------------
+# Prevent setuptools_scm issues if .git is missing
+ENV SETUPTOOLS_SCM_PRETEND_VERSION_FOR_GEOAI=0.0.0
+
+RUN rm -rf /home/jovyan/geoai/geoai.egg-info && \
+    pip install . && \
+    mkdir -p /home/jovyan/work && \
+    fix-permissions /home/jovyan
+
+# -------------------------------------------------------
+# 6. Set back to default user
+# -------------------------------------------------------
+WORKDIR /home/jovyan
+USER jovyan
+
+
+# -------------------------------------------------------
+# 7. Run the docker container
+# -------------------------------------------------------
+# docker run -it -p 8888:8888 -v $(pwd):/home/jovyan/work giswqs/geoai:latest

{geoai_py-0.5.6 → geoai_py-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geoai-py
-Version: 0.5.6
+Version: 0.6.0
 Summary: A Python package for using Artificial Intelligence (AI) with geospatial data
 Author-email: Qiusheng Wu <giswqs@gmail.com>
 License: MIT License
@@ -26,6 +26,7 @@ Requires-Dist: leafmap
 Requires-Dist: localtileserver
 Requires-Dist: mapclassify
 Requires-Dist: maplibre
+Requires-Dist: opencv-python-headless
 Requires-Dist: overturemaps
 Requires-Dist: planetary-computer
 Requires-Dist: pystac-client

geoai_py-0.6.0/docs/examples/samgeo.ipynb

@@ -0,0 +1,190 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Semantic Segmentation with SAM and GeoAI\n",
+    "\n",
+    "[![image](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/opengeos/geoai/blob/main/docs/examples/samgeo.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\n",
+    "from geoai.sam import SamGeo"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Download sample data"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "url = (\n",
+    "    \"https://huggingface.co/datasets/giswqs/geospatial/resolve/main/sam_demo_image.tif\"\n",
+    ")\n",
+    "image = geoai.download_file(url)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Visualize data"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "geoai.view_raster(url)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Initialize model"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "sam = SamGeo(model=\"facebook/sam-vit-huge\", automatic=True)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Automatic mask generation"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "sam.generate(source=image, output=\"masks.tif\", foreground=True)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Visualize results"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "sam.show_masks()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "sam.show_anns(alpha=0.6)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "geoai.view_raster(\"masks.tif\", colormap=\"tab20\", opacity=0.6, basemap=url)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "geoai.create_split_map(\n",
+    "    left_layer=\"masks.tif\",\n",
+    "    right_layer=url,\n",
+    "    left_args={\n",
+    "        \"colormap\": \"tab20\",\n",
+    "        \"opacity\": 1,\n",
+    "    },\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "![image](https://github.com/user-attachments/assets/8735160b-efb3-4b3c-9b1f-f83188874585)"
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "default_lexer": "ipython3"
+  },
+  "kernelspec": {
+   "display_name": "Python 3",
+   "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.2"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 4
+}

geoai_py-0.6.0/docs/sam.md

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

{geoai_py-0.5.6 → geoai_py-0.6.0}/geoai/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = """Qiusheng Wu"""
 __email__ = "giswqs@gmail.com"
-__version__ = "0.5.6"
+__version__ = "0.6.0"
 
 
 import os

{geoai_py-0.5.6 → geoai_py-0.6.0}/geoai/classify.py

@@ -1,6 +1,7 @@
 """The module for training semantic segmentation models for classifying remote sensing imagery."""
 
 import os
+
 import numpy as np
 
 
@@ -110,23 +111,24 @@ def train_classifier(
     Returns:
         object: Trained SemanticSegmentationTask model.
     """
+    import multiprocessing as mp
+    import timeit
+
+    import albumentations as A
     import lightning.pytorch as pl
+    import torch
+    from lightning.pytorch.callbacks import ModelCheckpoint
+    from lightning.pytorch.loggers import CSVLogger
     from torch.utils.data import DataLoader
-    from torchgeo.datasets import stack_samples, RasterDataset
+    from torchgeo.datamodules import GeoDataModule
+    from torchgeo.datasets import RasterDataset, stack_samples
     from torchgeo.datasets.splits import random_bbox_assignment
     from torchgeo.samplers import (
-        RandomGeoSampler,
-        RandomBatchGeoSampler,
         GridGeoSampler,
+        RandomBatchGeoSampler,
+        RandomGeoSampler,
     )
-    import torch
-    import multiprocessing as mp
-    import timeit
-    import albumentations as A
-    from torchgeo.datamodules import GeoDataModule
     from torchgeo.trainers import SemanticSegmentationTask
-    from lightning.pytorch.callbacks import ModelCheckpoint
-    from lightning.pytorch.loggers import CSVLogger
 
     # Create a wrapper class for albumentations to work with TorchGeo format
     class AlbumentationsWrapper:
@@ -376,21 +378,19 @@ def _classify_image(
     Returns:
         str: Path to the saved classified image.
     """
-    import numpy as np
     import timeit
-    from tqdm import tqdm
 
+    import numpy as np
+    import rasterio
     import torch
+    from rasterio.io import MemoryFile
+    from rasterio.merge import merge
+    from rasterio.transform import from_origin
     from torch.utils.data import DataLoader
-
     from torchgeo.datasets import RasterDataset, stack_samples
     from torchgeo.samplers import GridGeoSampler
     from torchgeo.trainers import SemanticSegmentationTask
-
-    import rasterio
-    from rasterio.transform import from_origin
-    from rasterio.io import MemoryFile
-    from rasterio.merge import merge
+    from tqdm import tqdm
 
     # Set default output path if not provided
     if output_path is None:
@@ -619,14 +619,12 @@ def classify_image(
         str: Path to the saved classified image.
     """
     import timeit
-
-    import torch
-    from torchgeo.trainers import SemanticSegmentationTask
+    import warnings
 
     import rasterio
-
-    import warnings
+    import torch
     from rasterio.errors import NotGeoreferencedWarning
+    from torchgeo.trainers import SemanticSegmentationTask
 
     # Disable specific GDAL/rasterio warnings
     warnings.filterwarnings("ignore", category=UserWarning, module="rasterio._.*")
@@ -863,9 +861,10 @@ def classify_images(
         list: List of paths to the saved classified images.
     """
     # Import required libraries
-    from tqdm import tqdm
     import glob
 
+    from tqdm import tqdm
+
     # Process directory input
     if isinstance(image_paths, str) and os.path.isdir(image_paths):
         # Set default output directory if not provided

{geoai_py-0.5.6 → geoai_py-0.6.0}/geoai/geoai.py

@@ -7,21 +7,21 @@ logging.getLogger("maplibre").setLevel(logging.ERROR)
 import leafmap
 import leafmap.maplibregl as maplibregl
 
+from .classify import classify_image, classify_images, train_classifier
 from .download import (
     download_naip,
     download_overture_buildings,
     download_pc_stac_item,
-    get_overture_data,
     extract_building_stats,
+    get_overture_data,
     pc_collection_list,
     pc_item_asset_list,
-    pc_stac_search,
     pc_stac_download,
+    pc_stac_search,
     read_pc_item_asset,
     view_pc_item,
     view_pc_items,
 )
-from .classify import train_classifier, classify_image, classify_images
 from .extract import *
 from .hf import *
 from .segment import *