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

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

@@ -138,3 +138,4 @@ ENV/
 .vscode/
 .DS_Store
 CLAUDE.md
+lightning_logs/

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geoai-py
-Version: 0.27.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

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

@@ -2,7 +2,7 @@
 
 __author__ = """Qiusheng Wu"""
 __email__ = "giswqs@gmail.com"
-__version__ = "0.27.0"
+__version__ = "0.28.0"
 
 
 import os
@@ -106,6 +106,8 @@ from .utils import (
     regularization,
     hybrid_regularization,
     adaptive_regularization,
+    flipnslide_augmentation,
+    export_flipnslide_tiles,
 )
 
 from .timm_train import (
@@ -170,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 (

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

@@ -1717,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:
@@ -1879,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:

{geoai_py-0.27.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.27.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: