From 3275f79c74ba47abfec935a8fb656af96ca01bfc Mon Sep 17 00:00:00 2001 From: wasserth Date: Tue, 9 Apr 2024 13:14:22 +0200 Subject: [PATCH] round pi_times to 2 decimals --- totalsegmentator/bin/totalseg_get_phase.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/totalsegmentator/bin/totalseg_get_phase.py b/totalsegmentator/bin/totalseg_get_phase.py index fa63c73a5..6fa1a77ac 100644 --- a/totalsegmentator/bin/totalseg_get_phase.py +++ b/totalsegmentator/bin/totalseg_get_phase.py @@ -84,7 +84,7 @@ def get_ct_contrast_phase(ct_img: nib.Nifti1Image): phase, probability = pi_time_to_phase(pi_time) return {"pi_time": pi_time, "phase": phase, "probability": probability, - "pi_time_min": float(preds.min()), "pi_time_max": float(preds.max())} + "pi_time_min": round(float(preds.min()), 2), "pi_time_max": round(float(preds.max()), 2)} def main():