-
Notifications
You must be signed in to change notification settings - Fork 1.5k
Test time augmentation need to update with new features #3732
Copy link
Copy link
Closed
Labels
enhancementNew feature or requestNew feature or request
Milestone
Description
Is your feature request related to a problem? Please describe.
Hi @rijobro ,
Thanks for providing the TTA module since MONAI 0.5.
@dongyang0122 @holgerroth and I discussed the TTA module for evaluation cases, and found several things should be updated:
- All the invert transform logic in TTA can be replaced by
Invertdtransform directly now.
AndInvertdtransform provides more features. - The post transform example in the doc-string is not
decollatelogic. - Better to ensure
determinismfor the random transforms in the example of doc-string. - It copy between Numpy and Tensor several times because we didn't support Tensor in transforms and don't have the proper numpy <-> tensor conversion before:
https://github.com/Project-MONAI/MONAI/blob/dev/monai/data/test_time_augmentation.py#L188
https://github.com/Project-MONAI/MONAI/blob/dev/monai/data/test_time_augmentation.py#L190
It will also cause numpy error if the transforms inverse() outputs Tensor.
Should be updated to supportNdarrayOrTensordirectly. - Some doc-strings exceed 120 chars.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request