Change how TTA works!


For the problem that I am currently working on, I don’t want TTA to look into four corners or perform cropping. How can I change it ?

I want TTA to use only those augmentations that I used on training data.

You need to change this line according to your needs.

augm_tfm = [o for o in if o.tfm not in
               (crop_pad, flip_lr, dihedral, zoom)]

Rest copy all the code from the file.

1 Like

Thank you for the answer! I will give it a try :slight_smile: