diff --git a/src/mednet/libs/segmentation/scripts/predict.py b/src/mednet/libs/segmentation/scripts/predict.py index 84c50d47f3e109124bfcadef29a15557ad230aa6..db43e0e1172e66fb05a72d05c34c711318d2c43a 100644 --- a/src/mednet/libs/segmentation/scripts/predict.py +++ b/src/mednet/libs/segmentation/scripts/predict.py @@ -67,3 +67,4 @@ def predict( json_predictions = run(model, datamodule, device_manager, output_folder) save_json_with_backup(output_folder / "predictions.json", json_predictions) + logger.info(f"Predictions saved to `{str(json_predictions)}`")