Skip to main content

The World-Class Geospatial AI Platform — standalone 119-model registry, DINOv3, Prithvi-EO-2.0, FastAPI serving, edge/cloud deploy, 51 pipelines, 25 notebooks

Project description

PyGeoVision

image image image Tests PyGeoFetch GeoAI DOI

A production-ready Python platform unifying satellite data acquisition and geospatial AI —
bridging PyGeoFetch (22+ providers) and GeoAI (full AI stack) in one coherent API.


📖 Introduction

PyGeoVision is a world-class geospatial AI platform that brings together two exceptional open-source packages under a single, unified API. It delegates all satellite data operations to PyGeoFetch — providing access to 22+ providers including Sentinel, Landsat, Planet, Maxar, Copernicus, USGS, and more — and all AI operations to GeoAI, exposing its full 24-subsystem stack covering segmentation, detection, classification, change detection, SAM, foundation models, embeddings, cloud masking, super-resolution, and ONNX export.

PyGeoVision's design principle is integration, not reimplementation. It is the bridge layer that makes PyGeoFetch and GeoAI work seamlessly together, adding 10 end-to-end pipelines, a full CLI, experiment tracking, distributed training, automated labeling, and YAML pipeline orchestration on top.

The package provides six core capabilities:

  1. Authenticated search and download of satellite imagery from 22+ providers with caching, parallel downloads, and post-processing chains.
  2. End-to-end pipelines that wire PyGeoFetch data directly into GeoAI inference with a single function call.
  3. A complete AI training stack with 14 model architectures, 6 specialist losses, and distributed GPU support.
  4. Automated dataset labeling from 7 sources including OSM, Microsoft, Google, ESA WorldCover, SAM, and foundation models.
  5. A full CLI (pygeovision) covering data, AI, pipelines, and model management.
  6. YAML pipeline orchestration for scheduled, repeatable geospatial workflows.

📝 Statement of Need

Applying AI to geospatial data requires navigating fragmented ecosystems — separate tools for data acquisition, preprocessing, model training, and inference — leading to steep learning curves, brittle pipelines, and reproducibility challenges. Existing packages like TorchGeo and TerraTorch provide excellent foundational tools but leave the data acquisition layer largely unsolved. PyGeoFetch addresses data acquisition comprehensively, and GeoAI addresses the AI layer comprehensively, but combining them into production workflows requires significant integration work.

PyGeoVision fills this gap by providing a unified, high-level interface that:

  • Gives geospatial researchers a single import to go from satellite search to AI inference.
  • Gives AI practitioners streamlined access to 22 satellite data providers without managing APIs, authentication, and file formats manually.
  • Gives organizations a production-ready platform with CLI tooling, YAML pipelines, scheduling, and experiment tracking.

With 10 built-in end-to-end pipelines covering building footprints, change detection, land cover, water bodies, solar detection, crop monitoring, disaster assessment, deforestation, urban growth, and carbon estimation, PyGeoVision dramatically reduces the time from raw satellite imagery to actionable geospatial intelligence.


Citations

If you find PyGeoVision useful in your research, please consider citing the following works:

@article{Wu2026geoai,
  author  = {Wu, Qiusheng},
  title   = {GeoAI: A Python package for integrating artificial intelligence with geospatial data analysis and visualization},
  journal = {Journal of Open Source Software},
  year    = {2026},
  volume  = {11},
  number  = {118},
  pages   = {9605},
  doi     = {10.21105/joss.09605}
}

🚀 Key Features

🛰️ Satellite Data — 22 Providers via PyGeoFetch

  • Unified search and download across Sentinel, Landsat, Planet, Maxar, Airbus, USGS, Copernicus, NASA, JAXA, and more
  • Secure credential management via system keyring (API keys, OAuth2, user/password)
  • Parallel downloads with checksum verification, resume support, and bandwidth throttling
  • Post-processing chains: unzip → reproject → compress → NDVI/NDWI → Cloud Optimized GeoTIFF
  • YAML pipeline orchestration with cron scheduling

🤖 AI Inference — 24 Subsystems via GeoAI

  • Segmentation: buildings, solar panels, agriculture fields, water bodies, custom models
  • Detection: cars, ships, parking spots, natural-language grounded detection (GroundedSAM), RF-DETR
  • Classification: scene classification, CLIP zero-shot land cover, batch inference
  • Change detection: ChangeSTAR bi-temporal change detection
  • Foundation models: NASA Prithvi, SAM, DINOv3, Tessera satellite embeddings
  • Cloud masking, super-resolution (ESRGAN), ONNX export, canopy height estimation

⚙️ End-to-End Pipelines (10)

Pipeline Description
building_footprints Sentinel-2 / NAIP → GeoAI BuildingFootprintExtractor → GeoJSON
change_detection Bi-temporal Sentinel-2 → ChangeSTAR → change mask
land_cover Sentinel-2 → SegFormer / ESA WorldCover → classification map
water_bodies Sentinel-2 → NDWI segmentation → water polygons
solar_detection NAIP / Sentinel-2 → SolarPanelDetector → GeoJSON
crop_monitoring Seasonal Sentinel-2 stack → crop type map
disaster_assessment Post-event imagery → Siamese-UNet → damage assessment
deforestation Bi-temporal Landsat/S2 → ChangeFormer → forest loss mask
urban_growth Bi-temporal Landsat → Siamese-UNet → urban expansion map
carbon_estimation Sentinel-2 NDVI → AGB formula → carbon stock estimate

🧠 Own AI Training Stack

  • 14 model architectures: U-Net, SegFormer, DeepLabV3+, FCOS, RetinaNet, ViT, ChangeFormer, ESRGAN, and more
  • GeoTrainer with 6 specialist losses (Dice, Focal, Tversky, Unified Focal, Weighted CE, Change Detection)
  • Distributed multi-GPU training, mixed precision, gradient accumulation
  • TiledInference with Gaussian blending for large-scene inference
  • ExperimentTracker and DriftDetector for production monitoring

🏷️ Automated Labeling (7 Sources)

OpenStreetMap · Microsoft Global Buildings · Google Open Buildings · ESA WorldCover · Google Dynamic World · SAM auto-labeling · Foundation model labeling


📦 Installation

# Core — data + basic inference
pip install pygeovision

# + Geospatial processing (rasterio, geopandas, rioxarray)
pip install "pygeovision[geo]"

# + Training stack (PyTorch, SMP, transformers, timm)
pip install "pygeovision[train]"

# + Foundation models (DINOv3, Prithvi-EO-2.0)
pip install "pygeovision[foundation]"

# + Vision-language models (CLIP, Moondream)
pip install "pygeovision[vlm]"

# + Time-series analysis
pip install "pygeovision[timeseries]"

# + Serving API (FastAPI, uvicorn, websockets)
pip install "pygeovision[serve]"

# + Everything
pip install "pygeovision[all]"

Requirements: Python 3.10+ · PyGeoFetch · GeoAI (optional) · PyTorch 2.0+


⚡ Quick Start

import warnings; warnings.filterwarnings('ignore')
import pathlib, json
import numpy as np
import matplotlib.pyplot as plt
import pygeovision as pgv
from pygeovision.models import get_model
from pygeovision.inference.tiled import TiledInference

client = pgv.PyGeoVision()

BBOX       = (-0.15, 51.47, -0.1, 51.52)
51.51191,-0.23590
DATE_RANGE = ('2024-06-01', '2024-08-31')
PROVIDERS  = ['planetary_computer']

DATA_DIR = pathlib.Path('./results/notebook')
DATA_DIR.mkdir(parents=True, exist_ok=True)

results = client.search(
    bbox            = BBOX,
    date_range      = DATE_RANGE,
    providers       = PROVIDERS,
    cloud_cover_max = 10,
)
print(f"Found {len(results)} scenes")
for r in results[:10]:
    print(f"  {r.provider:<22} {r.datetime[:10]}  cloud={r.cloud_cover:.1f}%  {r.id[:40]}")


BANDS = ['B02', 'B03', 'B04', 'B08', 'B8A', 'B11', 'B12']

downloads = client.download(
    results[:1],
    output_dir   = str(DATA_DIR),
    bands        = BANDS,
    post_process = ['reproject:EPSG:32630', 'cog'],
)
scene_path = downloads[0].path if downloads and downloads[0].success else None
scl_cands  = list(DATA_DIR.rglob('*SCL*.tif')) + list(DATA_DIR.rglob('*scl*.tif'))
scl_path   = str(scl_cands[0]) if scl_cands else None
if scene_path:
    d = downloads[0]
    print(f"Downloaded   : {scene_path}")
    print(f"Size         : {d.bytes_downloaded/1024/1024:.1f} MB")
    print(f"SCL mask     : {scl_path}")
else:
    print("Download failed or scene unavailable — check provider availability")


PREPROCESSED = DATA_DIR / 'london_preprocessed.tif'

boundary = {
  "type": "FeatureCollection",
  "features": [
    {
      "type": "Feature",
      "properties": {
        "name": "Study Area",
        "center_lat": 52.00669,
        "center_lon": -1.02688
      },
      "geometry": {
        "type": "Polygon",
        "coordinates": [[
          [-1.04188, 52.01669],
          [-1.03488, 52.02169],
          [-1.02188, 52.01869],
          [-1.00888, 52.01169],
          [-1.01488, 52.00169],
          [-1.02388, 51.99169],
          [-1.03488, 51.99469],
          [-1.04488, 52.00169],
          [-1.04188, 52.01669]
        ]]
      }
    }
  ]
}
# BBOX = (-0.25, 51.50, -0.20, 51.53)
if scene_path:
    ready = client.prepare_for_ai(
        scene_path,
        stack_bands  = BANDS,
        # bbox         = BBOX,
        bbox_crs     = 'EPSG:32630',
        scl_path     = scl_path,
        clip_geojson= boundary,
        scl_keep_classes = [4,5,6],
        normalise    = 'scale_factor',
        scale_factor = 10000.0,
        model_type   = 'segmentation',
        output_path  = str(PREPROCESSED),
    )
    print("Preprocessing steps :", ready['steps'])
    print("Output shape (C,H,W):", ready['shape'])
    print("Resolution          :", ready['resolution_m'], "m")
    print("Validation          :", "PASSED" if ready['report'] and ready['report'].passed else "FIXED (auto)")
    arr = ready['array']
    print(f"Value range         : {arr.min():.4f}{arr.max():.4f}")
    print(f"NaN count           : {np.isnan(arr).sum()}")
else:
    print("No scene available — cannot preprocess")
    ready = None


from pathlib import Path
from pygeovision.preprocess import Preprocessor

pre = Preprocessor()

# Set your data directory
DATA_DIR = Path("/home/mrtenkorang/pygeovision_v2/projects/results/notebook/planetary_computer")

# Build full paths to band files (files are directly in DATA_DIR)
band_files = [
    str(DATA_DIR / "T30UXC_20240823T110621_B02_10m_EPSG_32630.tif"),
    str(DATA_DIR / "T30UXC_20240823T110621_B03_10m_EPSG_32630.tif"),
    str(DATA_DIR / "T30UXC_20240823T110621_B04_10m_EPSG_32630.tif"),
    str(DATA_DIR / "T30UXC_20240823T110621_B08_10m_EPSG_32630.tif"),
    str(DATA_DIR / "T30UXC_20240823T110621_B11_20m_EPSG_32630.tif"),
    str(DATA_DIR / "T30UXC_20240823T110621_B12_20m_EPSG_32630.tif"),
]

# Stack the bands
pre.stack_bands(
    band_files,
    output_path=str(DATA_DIR / "sentinel2_6band.tif"),
    band_names=["Blue", "Green", "Red", "NIR", "SWIR1", "SWIR2"],
)
print("✅ Stacking complete!")

📋 Documentation

Comprehensive documentation is available at https://appiahkubis14.github.io/pygeovision-docs/, including:

  • Full API reference
  • Tutorials and example notebooks
  • Pipeline configuration guides
  • Contributing guide

🤝 Contributing

Contributions of all kinds are welcome. See our contributing guide for ways to get started.


📄 License

PyGeoVision is free and open source software, licensed under the Apache 2.0 License.


Acknowledgements

PyGeoVision is built on top of two exceptional open-source projects:

  • PyGeoFetch — Universal satellite data pipeline. PyGeoVision delegates all data search, download, authentication, caching, and pipeline orchestration to PyGeoFetch.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pygeovision-2.0.7.tar.gz (448.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

pygeovision-2.0.7-py3-none-any.whl (485.3 kB view details)

Uploaded Python 3

File details

Details for the file pygeovision-2.0.7.tar.gz.

File metadata

  • Download URL: pygeovision-2.0.7.tar.gz
  • Upload date:
  • Size: 448.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.15

File hashes

Hashes for pygeovision-2.0.7.tar.gz
Algorithm Hash digest
SHA256 2996f9f57670d5b508e3fa918cac4b02fd8aec00277d37bc502025a9a6ac9a6a
MD5 f6eb7e2e6a8e36698c41b36d3f6e01ae
BLAKE2b-256 92612f5dc57c277c69dc75168236b56064b653003a863f10c5f30cccd67f8e7f

See more details on using hashes here.

File details

Details for the file pygeovision-2.0.7-py3-none-any.whl.

File metadata

  • Download URL: pygeovision-2.0.7-py3-none-any.whl
  • Upload date:
  • Size: 485.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.15

File hashes

Hashes for pygeovision-2.0.7-py3-none-any.whl
Algorithm Hash digest
SHA256 4fe9501805b752f270d71b1b5ffe2cf826e5bd91124bf155a2de61cf98b3579a
MD5 69414bd897cfaf1bcded916bc9065ebd
BLAKE2b-256 f76418231bf15f8d159ff1683eb28d655616a3d5953c643841b05b278c42b13a

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page