Currently monai.data.TestTimeAugmentation seemingly assumes the inverse transformations can be applied to the predictions, which necessarily means the predictions must be spatial.
Seems fairly straightforward - would be willing to write up a PR if there's interest and no reason this doesn't make sense.
Currently
monai.data.TestTimeAugmentationseemingly assumes the inverse transformations can be applied to the predictions, which necessarily means the predictions must be spatial.Seems fairly straightforward - would be willing to write up a PR if there's interest and no reason this doesn't make sense.