TTA介绍
Test Time Augmentation(TTA),测试数据增强,是在测试阶段时,将输入的测试数据进行,翻转、旋转操作等数据增强,并最后对同一样本的不同数据增强的结果根据任务需求进行例如平均,求和等数据处理。
TTA实现
这里推荐github上一个库https://github.com/qubvel/ttach,可以直接调用tta,非常方便。
tta_model = tta.ClassificationTTAWrapper(model, tta.aliases.five_crop_transform(), merge_mode='mean')