geoai-py 0.14.0__tar.gz → 0.16.0__tar.gz

{geoai_py-0.14.0 → geoai_py-0.16.0}/.editorconfig

@@ -1,5 +1,6 @@
 # http://editorconfig.org
 
+# top-most EditorConfig file
 root = true
 
 [*]

{geoai_py-0.14.0 → geoai_py-0.16.0}/.gitignore

@@ -7,6 +7,9 @@ private/
 tests/data/
 CLAUDE.md
 AI_AGENTS.md
+docs/examples/timm_output/
+docs/examples/timm_seg_output/
+docs/examples/timm_buildings/
 
 # C extensions
 *.so

{geoai_py-0.14.0 → geoai_py-0.16.0}/.pre-commit-config.yaml

@@ -11,7 +11,7 @@ repos:
           - id: check-added-large-files
             args: ["--maxkb=500"]
 
-    - repo: https://github.com/psf/black
+    - repo: https://github.com/psf/black-pre-commit-mirror
       rev: 25.9.0
       hooks:
           - id: black-jupyter

geoai_py-0.16.0/CITATION.cff

@@ -0,0 +1,11 @@
+cff-version: 1.1.0
+message: "If you use this software, please cite it as below."
+authors:
+    - family-names: Wu
+      given-names: Qiusheng
+      orcid: https://orcid.org/0000-0001-5437-4073
+title: "GeoAI: A Python package for integrating artificial intelligence with geospatial data analysis and visualization"
+version: 0.14.0
+doi: 10.21105/joss.09025
+date-released: 2025-10-05
+url: "https://github.com/opengeos/goeai"

{geoai_py-0.14.0 → geoai_py-0.16.0}/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2023, Qiusheng Wu
+Copyright (c) 2023-2025, Qiusheng Wu
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
@@ -19,4 +19,3 @@ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
-

{geoai_py-0.14.0 → geoai_py-0.16.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geoai-py
-Version: 0.14.0
+Version: 0.16.0
 Summary: A Python package for using Artificial Intelligence (AI) with geospatial data
 Author-email: Qiusheng Wu <giswqs@gmail.com>
 License: MIT License
@@ -38,6 +38,7 @@ Requires-Dist: rasterio
 Requires-Dist: rioxarray
 Requires-Dist: scikit-image
 Requires-Dist: scikit-learn
+Requires-Dist: timm
 Requires-Dist: torch
 Requires-Dist: torchgeo
 Requires-Dist: torchinfo
@@ -46,6 +47,7 @@ Requires-Dist: transformers
 Provides-Extra: extra
 Requires-Dist: overturemaps; extra == "extra"
 Requires-Dist: torchange; extra == "extra"
+Requires-Dist: lightly-train; extra == "extra"
 Provides-Extra: agents
 Requires-Dist: strands-agents; extra == "agents"
 Requires-Dist: strands-agents-tools; extra == "agents"
@@ -92,6 +94,12 @@ GeoAI addresses this need by providing a unified, user-friendly interface that a
 
 The package's design philosophy emphasizes simplicity without sacrificing functionality, enabling users to perform sophisticated analyses such as building footprint extraction from satellite imagery, land cover classification, and change detection with just a few lines of code. By integrating cutting-edge AI models and providing seamless access to major geospatial data sources, GeoAI significantly lowers the barrier to entry for geospatial AI applications while maintaining the flexibility needed for advanced research applications.
 
+## Citations
+
+If you find GeoAI useful in your research, please consider citing the following paper to support my work. Thank you for your support.
+
+-   Wu, Q. (2025). GeoAI: A Python package for integrating artificial intelligence with geospatial data analysis and visualization. _Journal of Open Source Software_, 9025. [https://doi.org/10.21105/joss.09025](https://github.com/openjournals/joss-papers/blob/joss.09025/joss.09025/10.21105.joss.09025.pdf) (Under Review)
+
 ## 🚀 Key Features
 
 ### 📊 Advanced Geospatial Data Visualization

{geoai_py-0.14.0 → geoai_py-0.16.0}/README.md

@@ -36,6 +36,12 @@ GeoAI addresses this need by providing a unified, user-friendly interface that a
 
 The package's design philosophy emphasizes simplicity without sacrificing functionality, enabling users to perform sophisticated analyses such as building footprint extraction from satellite imagery, land cover classification, and change detection with just a few lines of code. By integrating cutting-edge AI models and providing seamless access to major geospatial data sources, GeoAI significantly lowers the barrier to entry for geospatial AI applications while maintaining the flexibility needed for advanced research applications.
 
+## Citations
+
+If you find GeoAI useful in your research, please consider citing the following paper to support my work. Thank you for your support.
+
+-   Wu, Q. (2025). GeoAI: A Python package for integrating artificial intelligence with geospatial data analysis and visualization. _Journal of Open Source Software_, 9025. [https://doi.org/10.21105/joss.09025](https://github.com/openjournals/joss-papers/blob/joss.09025/joss.09025/10.21105.joss.09025.pdf) (Under Review)
+
 ## 🚀 Key Features
 
 ### 📊 Advanced Geospatial Data Visualization

{geoai_py-0.14.0 → geoai_py-0.16.0}/geoai/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = """Qiusheng Wu"""
 __email__ = "giswqs@gmail.com"
-__version__ = "0.14.0"
+__version__ = "0.16.0"
 
 
 import os
@@ -101,3 +101,23 @@ def set_proj_lib_path(verbose=False):
 
 from .dinov3 import DINOv3GeoProcessor, analyze_image_patches, create_similarity_map
 from .geoai import *
+
+from .timm_train import (
+    get_timm_model,
+    modify_first_conv_for_channels,
+    TimmClassifier,
+    RemoteSensingDataset,
+    train_timm_classifier,
+    predict_with_timm,
+    list_timm_models,
+)
+
+from .timm_segment import (
+    TimmSegmentationModel,
+    SegmentationDataset,
+    train_timm_segmentation,
+    predict_segmentation,
+    train_timm_segmentation_model,
+    timm_semantic_segmentation,
+    push_timm_model_to_hub,
+)

{geoai_py-0.14.0 → geoai_py-0.16.0}/geoai/change_detection.py

@@ -561,7 +561,15 @@ class ChangeDetection:
 
         return fig
 
-    def visualize_results(self, image1_path, image2_path, binary_path, prob_path):
+    def visualize_results(
+        self,
+        image1_path,
+        image2_path,
+        binary_path,
+        prob_path,
+        title1="Earlier Image",
+        title2="Later Image",
+    ):
         """Create enhanced visualization with probability analysis."""
 
         # Load data
@@ -594,11 +602,11 @@ class ChangeDetection:
 
         # Row 1: Original and overlays
         axes[0, 0].imshow(img1_crop)
-        axes[0, 0].set_title("2019 Image", fontweight="bold")
+        axes[0, 0].set_title(title1, fontweight="bold")
         axes[0, 0].axis("off")
 
         axes[0, 1].imshow(img2_crop)
-        axes[0, 1].set_title("2022 Image", fontweight="bold")
+        axes[0, 1].set_title(title2, fontweight="bold")
         axes[0, 1].axis("off")
 
         # Binary overlay
@@ -708,6 +716,8 @@ class ChangeDetection:
         image2_path,
         binary_path,
         prob_path,
+        title1="Earlier Image",
+        title2="Later Image",
         output_path="split_comparison.png",
     ):
         """Create a split comparison visualization showing before/after with change overlay."""
@@ -742,7 +752,7 @@ class ChangeDetection:
         # Create split comparison
         fig, ax = plt.subplots(1, 1, figsize=(15, 10))
 
-        # Create combined image - left half is 2019, right half is 2022
+        # Create combined image - left half is earlier, right half is later
         combined_img = np.zeros_like(img1)
         combined_img[:, : w // 2] = img1[:, : w // 2]
         combined_img[:, w // 2 :] = img2[:, w // 2 :]
@@ -763,7 +773,7 @@ class ChangeDetection:
         ax.text(
             w // 4,
             50,
-            "2019",
+            title1,
             fontsize=20,
             color="white",
             ha="center",
@@ -772,7 +782,7 @@ class ChangeDetection:
         ax.text(
             3 * w // 4,
             50,
-            "2022",
+            title2,
             fontsize=20,
             color="white",
             ha="center",

{geoai_py-0.14.0 → geoai_py-0.16.0}/geoai/geoai.py

@@ -32,6 +32,9 @@ from .train import (
     instance_segmentation,
     instance_segmentation_batch,
     instance_segmentation_inference_on_geotiff,
+    lightly_embed_images,
+    lightly_train_model,
+    load_lightly_pretrained_model,
     object_detection,
     object_detection_batch,
     semantic_segmentation,