Skip to content

MeanEnsemble to accumulate results in-place  #6366

New issue
New issue
Open
#8141
Open
MeanEnsemble to accumulate results in-place #6366
#8141
Labels
Feature request
@wyli

Description

@wyli
wyli
opened on Apr 14, 2023

Describe the bug

MONAI/monai/transforms/post/array.py

Lines 628 to 636 in 825b8db

img_ = self.get_stacked_torch(img)
if self.weights is not None:
self.weights = self.weights.to(img_.device)
shape = tuple(self.weights.shape)
for _ in range(img_.ndimension() - self.weights.ndimension()):
shape += (1,)
weights = self.weights.reshape(*shape)
img_ = img_ * weights / weights.mean(dim=0, keepdim=True)

the current implementation keeps all the predictions which take up unnecessary space compared with maintaining an output and updating it in-place.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Feature request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions