copy geo-location to stain overlay images work

.vscode/launch.json

@@ -0,0 +1,15 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Python Debugger: Current File",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "${file}",
+            "console": "integratedTerminal"
+        }
+    ]
+}

.vscode/settings.json

@@ -2,5 +2,14 @@
     "cSpell.words": [
         "stainmask",
         "stainoverlay"
-    ]
+    ],
+    "python.testing.unittestArgs": [
+        "-v",
+        "-s",
+        ".",
+        "-p",
+        "test_*.py"
+    ],
+    "python.testing.pytestEnabled": false,
+    "python.testing.unittestEnabled": true
 }

config.ini

@@ -1,25 +1,25 @@
 [Settings]
 image_path = /home/roboticslab/Downloads/2d_up_9_to_8_test/images
-output_directory = /home/roboticslab/Downloads/2d_up_9_to_8_test/images_out/2024-02-28_21-23-17
+output_directory = /home/roboticslab/Downloads/2d_up_9_to_8_test/images_out/2024-02-26_23-06-36
 output_dir_message = output_directory is auto-generated.
 
 [CrackSegmentation]
 config = config_bo.json
 model = saved/UperNet/01-17_12-55/best_model.pth
-mask_directory = /home/roboticslab/Downloads/2d_up_9_to_8_test/images_out/2024-02-28_21-23-17/crackmask
+mask_directory = /home/roboticslab/Downloads/2d_up_9_to_8_test/images_out/2024-02-26_23-06-36/crackmask
 mask_dir_message = mask_directory is auto-generated.
 
 [StainSegmentation]
 config = config_stain.json
 model = saved/StainNet/2024-02-26_00-05/best_model.pth
-mask_directory = /home/roboticslab/Downloads/2d_up_9_to_8_test/images_out/2024-02-28_21-23-17/stainmask
+mask_directory = /home/roboticslab/Downloads/2d_up_9_to_8_test/images_out/2024-02-26_23-06-36/stainmask
 mask_dir_message = mask_directory is auto-generated.
 
 [CrackOverlay]
-overlay_directory = /home/roboticslab/Downloads/2d_up_9_to_8_test/images_out/2024-02-28_21-23-17/crackoverlay/images
+overlay_directory = /home/roboticslab/Downloads/2d_up_9_to_8_test/images_out/2024-02-26_23-06-36/crackoverlay/images
 overlay_dir_message = overlay_directory is auto-generated.
 
 [StainOverlay]
-overlay_directory = /home/roboticslab/Downloads/2d_up_9_to_8_test/images_out/2024-02-28_21-23-17/stainoverlay/images
+overlay_directory = /home/roboticslab/Downloads/2d_up_9_to_8_test/images_out/2024-02-26_23-06-36/stainoverlay/images
 overlay_dir_message = overlay_directory is auto-generated.
 

copy_geo_exiftool.py

@@ -1,4 +1,5 @@
 #!/home/roboticslab/Developer/image2overlays/.conda/bin/python
+import os
 import subprocess
 import json
 
@@ -16,6 +17,7 @@ def get_geolocation_with_exiftool(image_path):
 def write_geolocation_with_exiftool(dest_image, latitude, longitude, altitude):
     command = [
         "exiftool",
+        "-overwrite_original",  # Modify in-place
         f"-GPSLatitude={latitude}",
         f"-GPSLongitude={longitude}",
         f"-GPSAltitude={altitude}",

copy_geolocation.py

@@ -2,38 +2,25 @@
 import argparse
 import piexif
 from PIL import Image
-
-
-def process_single_image(raw_image_path, mask_image_path):
-    """Copies geolocation (EXIF) data from a raw image to a corresponding mask image.
-
-    Args:
-        raw_image_path (str): Path to the raw image file.
-        mask_image_path (str): Path to the mask image file.
-    """
-
-    if os.path.exists(mask_image_path):
-        try:
-            with Image.open(raw_image_path) as raw_image:
-                exif_data = piexif.load(raw_image.info.get("exif", {}))
-
-            # Handle cases where no EXIF data exists
-            if exif_data:
-                with Image.open(mask_image_path) as mask_image:
-                    mask_image.save(mask_image_path, "JPEG", exif=piexif.dump(exif_data))
-                print(f"Geolocation copied to: {mask_image_path}")
-
-        except (piexif.InvalidImageDataError, OSError) as e:
-            print(f"Error processing images: {e}")
+from copy_geo_exiftool import process_single_image
 
 def main(raw_images_dir=None, mask_images_dir=None):
     # ... (argument parsing - same as before) ...
 
     for raw_image_name in os.listdir(raw_images_dir):
         raw_image_path = os.path.join(raw_images_dir, raw_image_name)
         # ... (construct mask_image_path - same as before) ...
+        # Assuming mask images have a similar naming convention
+        mask_image_name = raw_image_name.replace(
+            "raw", "mask"
+        )  # Example transformation
+        mask_image_path = os.path.join(mask_images_dir, mask_image_name)
+        if os.path.exists(mask_image_path):
+            process_single_image(raw_image_path, mask_image_path)
+        else:
+            print("Mask image not found. Skipping to the next image.")
+            # raise FileNotFoundError("Mask image not found at specified path.")
 
-        process_single_image(raw_image_path, mask_image_path)
 
 
 if __name__ == "__main__":

copy_geolocation_crack.py

@@ -11,13 +11,14 @@
 # Extract image_path and mask_directory from the config file
 image_path = config['Settings']['image_path']
 # Assuming you want to use CrackSegmentation's mask_directory for this example
-mask_directory = config['CrackSegmentation']['mask_directory']
-
+# mask_directory = config['CrackSegmentation']['mask_directory']
+overlay_directory = config['CrackOverlay']['overlay_directory']
 # The path to your copy_geolocation.py script
 copy_geolocation_script_path = 'copy_geolocation.py'
 
 # Call copy_geolocation.py with the extracted paths
 # subprocess.run(['python', copy_geolocation_script_path, image_path, mask_directory], check=True)
-call_in_conda_env(f"python {copy_geolocation_script_path} {image_path} {mask_directory}", "/home/roboticslab/Developer/image2overlays/.conda")
+# call_in_conda_env(f"python {copy_geolocation_script_path} {image_path} {mask_directory}", "/home/roboticslab/Developer/image2overlays/.conda")
+call_in_conda_env(f"python {copy_geolocation_script_path} {image_path} {overlay_directory}", "/home/roboticslab/Developer/image2overlays/.conda")
 
-log_message("Copying geolocation info to stain overlay...")
+log_message("Copying geolocation info to crack overlay...")

copy_geolocation_stain.py

@@ -0,0 +1,19 @@
+#!~/anaconda3/bin/python
+#Copy geolocation data to stain overlay images
+import configparser
+from main import log_message, call_in_conda_env
+
+# Load the configuration file
+config = configparser.ConfigParser()
+config.read('config.ini')
+
+# Extract image_path and overlay_directory from the config file
+image_path = config['Settings']['image_path']
+overlay_directory = config['StainOverlay']['overlay_directory']
+# The path to your copy_geolocation.py script
+copy_geolocation_script_path = 'copy_geolocation.py'
+
+# Call copy_geolocation.py with the extracted paths
+call_in_conda_env(f"python {copy_geolocation_script_path} {image_path} {overlay_directory}", "/home/roboticslab/Developer/image2overlays/.conda")
+
+log_message("Copying geolocation info to stain overlay...")

main.py

@@ -41,6 +41,9 @@ def main():
 
         log_message("Running crack overlay...")
         call_in_conda_env("python crackoverlay.py")
+
+        log_message("Copying geolocation info to crack overlay...")
+        call_in_conda_env("python copy_geolocation_crack.py")
 
     if process_stain:
         # Run stain related processing
@@ -50,8 +53,8 @@ def main():
         log_message("Running stain overlay...")
         call_in_conda_env("python stainoverlay.py")
 
-        log_message("Copying geolocation info to crack overlay...")
-        call_in_conda_env("python copy_geolocation_crack.py")
+        log_message("Copying geolocation info to stain overlay...")
+        call_in_conda_env("python copy_geolocation_stain.py")
 
     log_message("Script sequence completed.")