Change how TTA works!

Part 1 (2019)
1

Hi,

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.

2

You need to change this line according to your needs.

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

Rest copy all the code from the fastai.vision.tta.py file.

1 Like
3

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