Skip to content

GPU-Optimized AI for Geospatial Annotation and Visual Search Accelerating Geospatial Intelligence through Distillation, Segmentation, and GPU Optimization.

Notifications You must be signed in to change notification settings

intelav/GeoAccel-AI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

5 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

โšก GeoAccel-AI: GPU-Optimized Geospatial Annotation

Accelerating Geospatial Intelligence through distillation, segmentation, and proprietary embeddings.
Developed under the SvarAikyam AI / AI Fusion initiative, this project integrates deep visual distillation models, segmentation refinement, and GPU-accelerated feature search to automate large-scale annotation and object discovery in high-resolution (3m) satellite imagery.


๐ŸŒ Overview

GeoAccel-AI enables automated detection, labeling, and visual search across multi-band satellite datasets โ€” empowering geospatial workflows through AI-driven similarity and segmentation refinement.
The visual search can identify and classify diverse infrastructure and terrain categories with minimal human supervision.

Supported object classes:

Brick Kiln | STP | Solar Panel | Sheds | Metro Shed | Play Ground | Pond-1 | Pond-2

๐Ÿ”น Core Pipelines

Stage Description
Prototype Creation Extracts proprietary feature embeddings from annotated .json polygons (ground truth).
Batch Prototype Builder Builds class prototypes across all TIF + JSON pairs.
Auto-Annotation (GPU Optimized) Performs multi-scale window detection using embedding similarity search and segmentation-based refinement for detected regions.
Batch Detection Executes large-scale annotation runs across hundreds of satellite tiles, generating object-level metadata.
Model Training YOLOv8-based detection fine-tuned on discovered regions; used as a region proposal network (RPN) for refinement.
Final Visual Search Embedding-based similarity retrieval to identify query objects in unseen images.
Interactive Review UI Provides OpenCV-based annotation validation, class editing, and YOLO export tools.
                                     |

Yolo Model Training

Below are the metrics for a representative YOLOv8 model trained for 200 epochs.
Since the extracted embeddings contain a degree of variability, moderate precision is expected โ€” yet sufficient for rapid bootstrapping of geospatial datasets. Metrics Figure 1: YOLOv8 training loss curves over 200 epochs.

โš™๏ธ GPU Profiling & Agentic Optimization

Integrated with Nsight Systems, Nsight Compute, and Torch Profiler for kernel-level insights, enabling precise GPU workload tuning. Includes experimental support for Agentic GPU Optimization, orchestrating autotuning of CUDA kernels and Triton ops for real-time efficiency gains on *RTX 3060.

Initial Profiling Report Figure 2: Initial Profiling Report on RTX3060 (12GB).

Optimized Profiling Report Figure 3: Optimized Profiling Report on RTX3060 (12GB).


๐Ÿ–ผ๏ธ Results

Initial Annotation
Figure 4: Initial Annotation for Training.

Final Known Objects
Figure 5: Final Result โ€” Detected Known Objects in Unseen Image.

Unseen Objects
Figure 6: Final Result โ€” Detected Unseen Objects in Unseen Image.


๐Ÿง  Research Context

GeoAccel-AI represents a unified, GPU-optimized geospatial AI framework that fuses:

  • Visual distillation for semantic and texture representation
  • Segmentation refinement for pixel-level precision
  • Agentic profiling & kernel discovery for dynamic GPU efficiency

It serves as a bridge between remote sensing, AI-driven automation, and edge-to-cloud scalability, enabling faster Earth observation analytics and infrastructure monitoring.


๐Ÿ“„ Citation

@misc{geoaccel_ai_2025,
  title  = {GeoAccel-AI: GPU-Optimized Auto-Annotation for Satellite Imagery},
  author = {Atul Vaish},
  year   = {2025},
  url    = {https://github.com/intelav/GeoAccel-AI}
}

ยฉ 2025 SvarAikyam AI | AI Fusion โ€” Applied Research in GPU Optimization & Geospatial Intelligence

About

GPU-Optimized AI for Geospatial Annotation and Visual Search Accelerating Geospatial Intelligence through Distillation, Segmentation, and GPU Optimization.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages