From bae8f59a6d42f0264b083832e543bb671b07a918 Mon Sep 17 00:00:00 2001 From: staydelight Date: Fri, 11 Oct 2024 16:51:56 +0800 Subject: [PATCH] fix-issue-6366 Signed-off-by: staydelight --- monai/transforms/post/array.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/monai/transforms/post/array.py b/monai/transforms/post/array.py index c0235d0268..d7d3a6f0bf 100644 --- a/monai/transforms/post/array.py +++ b/monai/transforms/post/array.py @@ -694,7 +694,8 @@ def __call__(self, img: Sequence[NdarrayOrTensor] | NdarrayOrTensor) -> NdarrayO total_weight += weight out_pt /= total_weight - return post_convert(out_pt, img) + return self.post_convert(out_pt, img) + class VoteEnsemble(Ensemble, Transform): """