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
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:
- Authenticated search and download of satellite imagery from 22+ providers with caching, parallel downloads, and post-processing chains.
- End-to-end pipelines that wire PyGeoFetch data directly into GeoAI inference with a single function call.
- A complete AI training stack with 14 model architectures, 6 specialist losses, and distributed GPU support.
- Automated dataset labeling from 7 sources including OSM, Microsoft, Google, ESA WorldCover, SAM, and foundation models.
- A full CLI (
pygeovision) covering data, AI, pipelines, and model management. - 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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file pygeovision-2.1.6.tar.gz.
File metadata
- Download URL: pygeovision-2.1.6.tar.gz
- Upload date:
- Size: 623.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
ea6c49da4dce635f955c87510942fd95afedd8d32e42cb9cf9a8f17c9c9f4141
|
|
| MD5 |
f85c39c29e4518c4df5473e802dbb387
|
|
| BLAKE2b-256 |
b981f65bf728e87a14975bb145370787ce908377714762e0d9cb99de4d6c60d5
|
File details
Details for the file pygeovision-2.1.6-py3-none-any.whl.
File metadata
- Download URL: pygeovision-2.1.6-py3-none-any.whl
- Upload date:
- Size: 649.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
a815693f62979771c91c3c33c02d9fd0cf8c3f2a7bcc6cbb23115ab99c51e2ac
|
|
| MD5 |
ada43a9c88f8ceb450c064d76c4b3663
|
|
| BLAKE2b-256 |
d96eabea2cad75cfd1829c9afcf46a344ce2dadf71c0f7c96b232e93d29979ca
|