diff --git a/monai/handlers/utils.py b/monai/handlers/utils.py index 0cd31b89c2..481d5a1611 100644 --- a/monai/handlers/utils.py +++ b/monai/handlers/utils.py @@ -126,6 +126,7 @@ class mean median max 5percentile 95percentile notnans else: class_labels = [str(i) for i in class_labels] # ensure to have a list of str + class_labels += ["mean"] v = np.concatenate([v, np.nanmean(v, axis=1, keepdims=True)], axis=1)