Hi, I wonder if you can add some tools for test-time-augmentation, especially in semantic segmentation tasks. Thank you