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

{geoai_py-0.15.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.15.0 → geoai_py-0.16.0}/PKG-INFO

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

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

@@ -2,7 +2,7 @@
 
 __author__ = """Qiusheng Wu"""
 __email__ = "giswqs@gmail.com"
-__version__ = "0.15.0"
+__version__ = "0.16.0"
 
 
 import os

{geoai_py-0.15.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.15.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,