From 01a7fef9a2d4d38a54170943cad090b184cc6a19 Mon Sep 17 00:00:00 2001 From: Rohit7824567 Date: Thu, 19 Mar 2026 21:17:48 +0530 Subject: [PATCH] zip processed_images with predictions instead of valid_images Zipping valid_images (full list) with predictions (subset on resume) saved masks to wrong filenames in Alaska coastal batch inference. Using processed_images ensures correct mask-to-image pairing. --- training_pipeline/predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/training_pipeline/predict.py b/training_pipeline/predict.py index 27886f7..ed3e23d 100644 --- a/training_pipeline/predict.py +++ b/training_pipeline/predict.py @@ -313,7 +313,7 @@ def main(): checkpoint_path, resume) # Save predictions - for i, (image_path, pred_mask) in enumerate(zip(valid_images, predictions)): + for image_path, pred_mask in zip(processed_images, predictions): base_name = os.path.splitext(os.path.basename(image_path))[0] mask_save_path = f"{base_name}_predicted_mask.png" Image.fromarray(pred_mask * 255).save(mask_save_path)