From 51c9eb84ecd1b2f66e4c1fe7caf2849ec029eb88 Mon Sep 17 00:00:00 2001 From: Rohit7824567 Date: Thu, 19 Mar 2026 20:44:11 +0530 Subject: [PATCH] Return processed images along with predictions MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit fix: return processed_images alongside predictions from predict_batch Previously returned only `predictions`, causing zip misalignment in main() when resuming — masks were saved to wrong image filenames. - Before: `return predictions` - After: `return processed_images, predictions` Ensures correct image-mask pairing during Alaska coastline batch inference. --- 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..6961d5b 100644 --- a/training_pipeline/predict.py +++ b/training_pipeline/predict.py @@ -117,7 +117,7 @@ def predict_batch(model, image_paths, device="cuda", threshold=0.5, checkpoint_p save_checkpoint(checkpoint_path, processed_images, predictions, metadata) print(f"Final checkpoint saved: {len(processed_images)} images processed") - return predictions + return processed_images, predictions def visualize_prediction(image_path, pred_mask, save_path=None): """