So that would entail adding in a separate set of only inversible transforms for the test dl instead of utilizing the validation ones, modifying the test dl to capture the transform parameters, and then the tta function to do the inversion on the prediction?
Here was another thread on the topic:
And also, not all of them would need to be inversed, only those that change pixel locations. Crops probably wouldn’t work.