Skip to content

Adding evidence of performance improvement by using test-time augmentation (TTA) in the tutorial notebook #191

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
dongyang0122 opened this issue Apr 21, 2021 · 4 comments
Assignees
Labels
enhancement New feature or request

Comments

@dongyang0122
Copy link
Collaborator

Current tutorial notebook about test-time augmentation (TTA) simply shows how to apply it for brain MRI segmentation in 2D planes. However, it is unclear the if the TTA method would improve the segmentation quality in practice.
https://github.com/Project-MONAI/tutorials/blob/master/modules/inverse_transforms_and_test_time_augmentations.ipynb

It is better to show the accuracy and quality (visual) improvement of segmentation masks and let users know how to properly apply the TTA method in practice boosting the performance.

@Nic-Ma
Copy link
Contributor

Nic-Ma commented Apr 21, 2021

Hi @rijobro ,

Could you please help take a look when you are back?

Thanks.

@Nic-Ma Nic-Ma added the enhancement New feature or request label Apr 21, 2021
@rijobro
Copy link
Contributor

rijobro commented Apr 26, 2021

Hi @dongyang0122 are you saying you'd like a comparison between the inference of a single image versus the mode of multiple random instances computed with TTA?

@dongyang0122
Copy link
Collaborator Author

My point is to justify the usage of TTA in practice. It would be nice to show the segmentation improvement in terms of accuracy and appearance.

@ahatamiz
Copy link
Contributor

Hi @rijobro

I tried to use this TTA implementation for a segmentation application. After reading the source code, it turns out it expects raw input data dictionary as opposed to a PyTorch tensors ? ideally this is a post-transform which should also accept PyTorch tensors and then apply additional transforms.

I also agree with @dongyang0122. The tutorial can be improved by including a 3D use-case.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

4 participants