From 8bc86ef93bac37a9a1e3f9d44a99caee507b04f9 Mon Sep 17 00:00:00 2001 From: Rohit7824567 Date: Thu, 19 Mar 2026 21:15:20 +0530 Subject: [PATCH] unpack processed_images from predict_batch return value in main() After predict_batch was updated to return (processed_images, predictions), main() was still unpacking only predictions, causing ValueError during Alaska coastal batch inference runs. Unpacking both values ensures correct mask-to-image mapping when saving results after resumed or full batch prediction jobs --- training_pipeline/predict.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/training_pipeline/predict.py b/training_pipeline/predict.py index 27886f7..de623c8 100644 --- a/training_pipeline/predict.py +++ b/training_pipeline/predict.py @@ -309,8 +309,7 @@ def main(): print(f"Checkpoint path: {checkpoint_path}") print(f"Processing {len(valid_images)} images...") - predictions = predict_batch(model, valid_images, device, args.threshold, - checkpoint_path, resume) + processed_images, predictions = predict_batch(model, valid_images, device, args.threshold, checkpoint_path, resume) # Save predictions for i, (image_path, pred_mask) in enumerate(zip(valid_images, predictions)):