From f4992bd42621bac589fbd2ad1d7554870db81a58 Mon Sep 17 00:00:00 2001 From: kbressem Date: Fri, 10 Mar 2023 17:53:58 -0500 Subject: [PATCH 1/2] Add single if statement for warn Avoids crashing of the function if `ratios_[i] = 0` cannot be updated --- monai/transforms/utils.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/monai/transforms/utils.py b/monai/transforms/utils.py index ea3577159d..a5fbca32ee 100644 --- a/monai/transforms/utils.py +++ b/monai/transforms/utils.py @@ -569,8 +569,9 @@ def generate_label_classes_crop_centers( raise ValueError(f"ratios should not contain negative number, got {ratios_}.") for i, array in enumerate(indices): - if len(array) == 0 and warn: - warnings.warn(f"no available indices of class {i} to crop, set the crop ratio of this class to zero.") + if len(array) == 0: + if warn: + warnings.warn(f"no available indices of class {i} to crop, set the crop ratio of this class to zero.") ratios_[i] = 0 centers = [] From d92b4eeac4a0a444bd19d0bb5467d0d416fe83df Mon Sep 17 00:00:00 2001 From: kbressem Date: Fri, 10 Mar 2023 17:56:33 -0500 Subject: [PATCH 2/2] DCO Remediation Commit for kbressem I, kbressem , hereby add my Signed-off-by to this commit: f4992bd42621bac589fbd2ad1d7554870db81a58 Signed-off-by: kbressem --- monai/transforms/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monai/transforms/utils.py b/monai/transforms/utils.py index a5fbca32ee..111bfa1102 100644 --- a/monai/transforms/utils.py +++ b/monai/transforms/utils.py @@ -570,9 +570,9 @@ def generate_label_classes_crop_centers( for i, array in enumerate(indices): if len(array) == 0: + ratios_[i] = 0 if warn: warnings.warn(f"no available indices of class {i} to crop, set the crop ratio of this class to zero.") - ratios_[i] = 0 centers = [] classes = rand_state.choice(len(ratios_), size=num_samples, p=np.asarray(ratios_) / np.sum(ratios_))