TTAch
使用PyTorch进行图像测试时间增强!
与“数据增强”对训练集所做的类似,“测试时间增强”的目的是对测试图像进行随机修改。 因此,我们不会对训练后的模型仅显示一次常规的“干净”图像,而是将其多次显示。 然后,我们将平均每个对应图像的预测,并将其作为我们的最终猜测[ ]。
Input
| # input batch of images
/ / /|\ \ \ # apply augmentations (flips, rotation, scale, etc.)
| | | | | | | # pass augmented batches through model
| | | | | | | # reverse transfo
1