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): """