geoai-py 0.26.0__py2.py3-none-any.whl → 0.28.0__py2.py3-none-any.whl

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/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/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/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/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/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/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: