geoai-py 0.26.0__tar.gz → 0.28.0__tar.gz

{geoai_py-0.26.0 → geoai_py-0.28.0}/.gitignore

@@ -32,6 +32,9 @@ docs/workshops/**/*.jpg
 docs/workshops/**/*.png
 docs/examples/data/
 docs/workshops/data/
+docs/examples/ndvi_regression_output/
+docs/examples/timm_regression_output/
+docs/examples/ndvi_model/
 *.pth
 
 # Distribution / packaging
@@ -135,3 +138,4 @@ ENV/
 .vscode/
 .DS_Store
 CLAUDE.md
+lightning_logs/

{geoai_py-0.26.0 → geoai_py-0.28.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geoai-py
-Version: 0.26.0
+Version: 0.28.0
 Summary: A Python package for using Artificial Intelligence (AI) with geospatial data
 Author-email: Qiusheng Wu <giswqs@gmail.com>
 License: MIT License
@@ -17,10 +17,8 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: albumentations
-Requires-Dist: buildingregulariser
 Requires-Dist: contextily
 Requires-Dist: datasets>=3.0
-Requires-Dist: ever-beta
 Requires-Dist: geopandas
 Requires-Dist: huggingface_hub
 Requires-Dist: jupyter-server-proxy
@@ -31,7 +29,6 @@ Requires-Dist: maplibre
 Requires-Dist: opencv-python-headless
 Requires-Dist: overturemaps
 Requires-Dist: planetary-computer
-Requires-Dist: psutil
 Requires-Dist: pyarrow
 Requires-Dist: pystac-client
 Requires-Dist: rasterio
@@ -53,12 +50,17 @@ Requires-Dist: lightly-train; extra == "extra"
 Requires-Dist: multiclean; extra == "extra"
 Requires-Dist: omnicloudmask; extra == "extra"
 Requires-Dist: smoothify; extra == "extra"
+Provides-Extra: building
+Requires-Dist: buildingregulariser; extra == "building"
 Provides-Extra: agents
 Requires-Dist: strands-agents; extra == "agents"
 Requires-Dist: strands-agents-tools; extra == "agents"
 Requires-Dist: strands-agents[ollama]; extra == "agents"
 Requires-Dist: strands-agents[anthropic]; extra == "agents"
 Requires-Dist: strands-agents[openai]; extra == "agents"
+Provides-Extra: onnx
+Requires-Dist: onnx; extra == "onnx"
+Requires-Dist: onnxruntime; extra == "onnx"
 Provides-Extra: sr
 Requires-Dist: opensr-model; extra == "sr"
 Dynamic: license-file
@@ -71,7 +73,7 @@ Dynamic: license-file
 [![Conda Downloads](https://img.shields.io/conda/dn/conda-forge/geoai.svg)](https://anaconda.org/conda-forge/geoai)
 [![Conda Recipe](https://img.shields.io/badge/recipe-geoai-green.svg)](https://github.com/conda-forge/geoai-py-feedstock)
 [![image](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-[![image](https://img.shields.io/badge/YouTube-Tutorials-red)](https://tinyurl.com/GeoAI-Tutorials)
+[![image](https://img.shields.io/badge/YouTube-Tutorials-red)](https://www.youtube.com/playlist?list=PLAxJ4-o7ZoPcvENqwaPa_QwbbkZ5sctZE)
 [![QGIS](https://img.shields.io/badge/QGIS-plugin-orange.svg)](https://opengeoai.org/qgis_plugin)
 
 [![logo](https://raw.githubusercontent.com/opengeos/geoai/master/docs/assets/logo_rect.png)](https://github.com/opengeos/geoai/blob/master/docs/assets/logo.png)
@@ -91,7 +93,7 @@ The package provides five core capabilities:
 5. Interactive visualization through integration with [Leafmap](https://github.com/opengeos/leafmap/) and [MapLibre](https://github.com/eoda-dev/py-maplibregl).
 6. Seamless QGIS integration via a dedicated GeoAI plugin, enabling users to run AI-powered geospatial workflows directly within the QGIS desktop environment, without writing code.
 
-GeoAI addresses the growing demand for accessible AI tools in geospatial research by providing high-level APIs that abstract complex machine learning workflows while maintaining flexibility for advanced users. The package supports multiple data formats (GeoTIFF, JPEG2000,GeoJSON, Shapefile, GeoPackage) and includes automatic device management for GPU acceleration when available. With over 10 modules and extensive notebook examples, GeoAI serves as both a research tool and educational resource for the geospatial AI community.
+GeoAI addresses the growing demand for accessible AI tools in geospatial research by providing high-level APIs that abstract complex machine learning workflows while maintaining flexibility for advanced users. The package supports multiple data formats (GeoTIFF, JPEG2000, GeoJSON, Shapefile, GeoPackage) and includes automatic device management for GPU acceleration when available. With over 10 modules and extensive notebook examples, GeoAI serves as both a research tool and educational resource for the geospatial AI community.
 
 ## 📝 Statement of Need
 
@@ -130,7 +132,7 @@ If you find GeoAI useful in your research, please consider citing the following
 
 -   Integration with [PyTorch Segmentation Models](https://github.com/qubvel-org/segmentation_models.pytorch) for automatic feature extraction
 -   Specialized segmentation algorithms optimized for satellite and aerial imagery
--   Streamlined workflows for segmenting buildings, water bodies, wetlands,solar panels, etc.
+-   Streamlined workflows for segmenting buildings, water bodies, wetlands, solar panels, etc.
 -   Export capabilities to standard geospatial formats (GeoJSON, Shapefile, GeoPackage, GeoParquet)
 
 ### 🔍 Image Classification

{geoai_py-0.26.0 → geoai_py-0.28.0}/README.md

@@ -6,7 +6,7 @@
 [![Conda Downloads](https://img.shields.io/conda/dn/conda-forge/geoai.svg)](https://anaconda.org/conda-forge/geoai)
 [![Conda Recipe](https://img.shields.io/badge/recipe-geoai-green.svg)](https://github.com/conda-forge/geoai-py-feedstock)
 [![image](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-[![image](https://img.shields.io/badge/YouTube-Tutorials-red)](https://tinyurl.com/GeoAI-Tutorials)
+[![image](https://img.shields.io/badge/YouTube-Tutorials-red)](https://www.youtube.com/playlist?list=PLAxJ4-o7ZoPcvENqwaPa_QwbbkZ5sctZE)
 [![QGIS](https://img.shields.io/badge/QGIS-plugin-orange.svg)](https://opengeoai.org/qgis_plugin)
 
 [![logo](https://raw.githubusercontent.com/opengeos/geoai/master/docs/assets/logo_rect.png)](https://github.com/opengeos/geoai/blob/master/docs/assets/logo.png)
@@ -26,7 +26,7 @@ The package provides five core capabilities:
 5. Interactive visualization through integration with [Leafmap](https://github.com/opengeos/leafmap/) and [MapLibre](https://github.com/eoda-dev/py-maplibregl).
 6. Seamless QGIS integration via a dedicated GeoAI plugin, enabling users to run AI-powered geospatial workflows directly within the QGIS desktop environment, without writing code.
 
-GeoAI addresses the growing demand for accessible AI tools in geospatial research by providing high-level APIs that abstract complex machine learning workflows while maintaining flexibility for advanced users. The package supports multiple data formats (GeoTIFF, JPEG2000,GeoJSON, Shapefile, GeoPackage) and includes automatic device management for GPU acceleration when available. With over 10 modules and extensive notebook examples, GeoAI serves as both a research tool and educational resource for the geospatial AI community.
+GeoAI addresses the growing demand for accessible AI tools in geospatial research by providing high-level APIs that abstract complex machine learning workflows while maintaining flexibility for advanced users. The package supports multiple data formats (GeoTIFF, JPEG2000, GeoJSON, Shapefile, GeoPackage) and includes automatic device management for GPU acceleration when available. With over 10 modules and extensive notebook examples, GeoAI serves as both a research tool and educational resource for the geospatial AI community.
 
 ## 📝 Statement of Need
 
@@ -65,7 +65,7 @@ If you find GeoAI useful in your research, please consider citing the following
 
 -   Integration with [PyTorch Segmentation Models](https://github.com/qubvel-org/segmentation_models.pytorch) for automatic feature extraction
 -   Specialized segmentation algorithms optimized for satellite and aerial imagery
--   Streamlined workflows for segmenting buildings, water bodies, wetlands,solar panels, etc.
+-   Streamlined workflows for segmenting buildings, water bodies, wetlands, solar panels, etc.
 -   Export capabilities to standard geospatial formats (GeoJSON, Shapefile, GeoPackage, GeoParquet)
 
 ### 🔍 Image Classification

{geoai_py-0.26.0 → geoai_py-0.28.0}/geoai/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = """Qiusheng Wu"""
 __email__ = "giswqs@gmail.com"
-__version__ = "0.26.0"
+__version__ = "0.28.0"
 
 
 import os
@@ -101,6 +101,14 @@ def set_proj_lib_path(verbose=False):
 
 from .dinov3 import DINOv3GeoProcessor, analyze_image_patches, create_similarity_map
 from .geoai import *
+from .utils import (
+    orthogonalize,
+    regularization,
+    hybrid_regularization,
+    adaptive_regularization,
+    flipnslide_augmentation,
+    export_flipnslide_tiles,
+)
 
 from .timm_train import (
     get_timm_model,
@@ -122,6 +130,25 @@ from .timm_segment import (
     push_timm_model_to_hub,
 )
 
+from .timm_regress import (
+    PixelRegressionModel,
+    PixelRegressionDataset,
+    create_regression_tiles,
+    train_pixel_regressor,
+    predict_raster,
+    evaluate_regression,
+    plot_regression_comparison,
+    plot_scatter,
+    plot_training_history,
+    visualize_prediction,
+    plot_regression_results,
+    # Backward compatibility aliases
+    TimmRegressor,
+    RegressionDataset,
+    train_timm_regressor,
+    create_regression_patches,
+)
+
 # Import tools subpackage
 from . import tools
 
@@ -145,6 +172,18 @@ except ImportError:
     # super_resolution not available (missing dependency)
     pass
 
+# ONNX Runtime support
+try:
+    from .onnx import (
+        ONNXGeoModel,
+        export_to_onnx,
+        onnx_semantic_segmentation,
+        onnx_image_classification,
+    )
+except ImportError:
+    # ONNX not available (missing dependency)
+    pass
+
 # Moondream Vision Language Model
 try:
     from .moondream import (
@@ -167,6 +206,7 @@ except ImportError:
 try:
     from .prithvi import (
         PrithviProcessor,
+        get_available_prithvi_models,
         load_prithvi_model,
         prithvi_inference,
     )

{geoai_py-0.26.0 → geoai_py-0.28.0}/geoai/auto.py

@@ -24,7 +24,6 @@ Example:
 import os
 from typing import Any, Dict, List, Optional, Tuple, Union
 
-import cv2
 import geopandas as gpd
 import numpy as np
 import rasterio
@@ -877,6 +876,8 @@ class AutoGeoModel:
         **kwargs: Any,
     ) -> Dict[str, Any]:
         """Run tiled inference for large images."""
+        import cv2  # Lazy import to avoid QGIS opencv conflicts
+
         if data.ndim == 3:
             _, height, width = data.shape
         else:
@@ -1872,6 +1873,7 @@ def show_segmentation(
         >>> result = geoai.auto.semantic_segmentation("aerial.tif", output_path="seg.tif")
         >>> fig = show_segmentation("aerial.tif", result["mask"])
     """
+    import cv2  # Lazy import to avoid QGIS opencv conflicts
     import matplotlib.pyplot as plt
 
     img, _ = _load_image_for_display(source)
@@ -1941,6 +1943,7 @@ def show_depth(
         >>> result = geoai.auto.depth_estimation("aerial.tif", output_path="depth.tif")
         >>> fig = show_depth("aerial.tif", result["depth"])
     """
+    import cv2  # Lazy import to avoid QGIS opencv conflicts
     import matplotlib.pyplot as plt
 
     img, _ = _load_image_for_display(source)

{geoai_py-0.26.0 → geoai_py-0.28.0}/geoai/change_detection.py

@@ -3,7 +3,6 @@
 import os
 from typing import Any, Dict, List, Optional, Tuple, Union
 
-import cv2
 import matplotlib.pyplot as plt
 import numpy as np
 import rasterio
@@ -736,6 +735,7 @@ class ChangeDetection:
         output_path="split_comparison.png",
     ):
         """Create a split comparison visualization showing before/after with change overlay."""
+        import cv2  # Lazy import to avoid QGIS opencv conflicts
 
         # Load data
         with rasterio.open(image1_path) as src:

{geoai_py-0.26.0 → geoai_py-0.28.0}/geoai/detectron2.py

@@ -6,7 +6,6 @@ import os
 import warnings
 from typing import Dict, List, Optional, Tuple, Union
 
-import cv2
 import numpy as np
 import rasterio
 import torch
@@ -135,6 +134,8 @@ def detectron2_segment(
     Returns:
         Dict containing segmentation results and output file paths
     """
+    import cv2  # Lazy import to avoid QGIS opencv conflicts
+
     check_detectron2()
 
     # Load the model
@@ -315,6 +316,8 @@ def visualize_detectron2_results(
     Returns:
         Visualization image as numpy array
     """
+    import cv2  # Lazy import to avoid QGIS opencv conflicts
+
     check_detectron2()
 
     # Load the image

{geoai_py-0.26.0 → geoai_py-0.28.0}/geoai/extract.py

@@ -6,7 +6,6 @@ import time
 from typing import Any, Dict, Generator, List, Optional, Tuple, Union
 
 # Third-Party Libraries
-import cv2
 import geopandas as gpd
 import matplotlib.pyplot as plt
 import numpy as np
@@ -440,6 +439,7 @@ class ObjectDetector:
         Returns:
             List of polygons as lists of (x, y) coordinates
         """
+        import cv2  # Lazy import to avoid QGIS opencv conflicts
 
         # Get parameters from kwargs or use instance defaults
         simplify_tolerance = kwargs.get("simplify_tolerance", self.simplify_tolerance)
@@ -637,6 +637,8 @@ class ObjectDetector:
         Returns:
             GeoDataFrame with objects
         """
+        import cv2  # Lazy import to avoid QGIS opencv conflicts
+
         # Use class defaults if parameters not provided
         simplify_tolerance = (
             simplify_tolerance
@@ -1715,21 +1717,21 @@ class ObjectDetector:
                             confidence = None
                             try:
                                 confidence = row.confidence
-                            except:
+                            except Exception:
                                 pass
 
                             # Method 2: Try dictionary-style access
                             if confidence is None:
                                 try:
                                     confidence = row["confidence"]
-                                except:
+                                except Exception:
                                     pass
 
                             # Method 3: Try accessing by index from the GeoDataFrame
                             if confidence is None:
                                 try:
                                     confidence = gdf.iloc[idx]["confidence"]
-                                except:
+                                except Exception:
                                     pass
 
                             if confidence is not None:
@@ -1877,19 +1879,19 @@ class ObjectDetector:
                                 confidence = None
                                 try:
                                     confidence = row.confidence
-                                except:
+                                except Exception:
                                     pass
 
                                 if confidence is None:
                                     try:
                                         confidence = row["confidence"]
-                                    except:
+                                    except Exception:
                                         pass
 
                                 if confidence is None:
                                     try:
                                         confidence = visible_gdf.iloc[idx]["confidence"]
-                                    except:
+                                    except Exception:
                                         pass
 
                                 if confidence is not None:
@@ -2165,6 +2167,7 @@ class ObjectDetector:
         Returns:
             GeoDataFrame with car detections and confidence values
         """
+        import cv2  # Lazy import to avoid QGIS opencv conflicts
 
         def _process_single_component(
             component_mask: np.ndarray,

{geoai_py-0.26.0 → geoai_py-0.28.0}/geoai/hf.py

@@ -343,7 +343,7 @@ def mask_generation(
                 # Try to convert from tensor or other format if needed
                 try:
                     mask_data = np.array(mask_data)
-                except:
+                except Exception:
                     print(f"Could not convert mask at index {i} to numpy array")
                     continue
 
@@ -395,7 +395,7 @@ def mask_generation(
                 try:
                     mask_data = np.array(mask_result)
                     score = 1.0  # Default score
-                except:
+                except Exception:
                     print(f"Could not process mask at index {i}")
                     continue
 
@@ -404,7 +404,7 @@ def mask_generation(
                 if not isinstance(mask_data, np.ndarray):
                     try:
                         mask_data = np.array(mask_data)
-                    except:
+                    except Exception:
                         print(f"Could not convert mask at index {i} to numpy array")
                         continue
 

{geoai_py-0.26.0 → geoai_py-0.28.0}/geoai/moondream.py

@@ -1305,7 +1305,7 @@ class MoondreamGeo:
             try:
                 summary_result = self.model.query(question=summary_prompt)
                 combined_answer = summary_result.get("answer", "")
-            except:
+            except Exception:
                 # Fall back to concatenation if summarization fails
                 combined_answer = " ".join([ta["answer"] for ta in tile_answers])
         else:
@@ -1411,7 +1411,7 @@ class MoondreamGeo:
             try:
                 summary_result = self.model.query(question=summary_prompt)
                 combined_caption = summary_result.get("answer", "")
-            except:
+            except Exception:
                 # Fall back to concatenation if summarization fails
                 combined_caption = " ".join([tc["caption"] for tc in tile_captions])
         else: