在调用tta的时候由于我的模型输入有images和meta数据所以报错了,自己更改了下源码,能跑了但是精度低了一点点还有些问题。
1、修改wrappers
找到对应的文件miniconda3/lib/python3.8/site-packages/ttach/wrappers.py
接着按图修改前项传播函数
![](https://i-blog.csdnimg.cn/blog_migrate/b51316ef9f204347c696ff1b55dff105.png)
2、修改增强(选做)
找到文件miniconda3/lib/python3.8/site-packages/ttach/aliases.py,可以自己添加想要的增强
![](https://i-blog.csdnimg.cn/blog_migrate/a9d1865f0b985b5bffaa633ec316f3aa.png)
输出测试
tta_model = tta.ClassificationTTAWrapper(model, tta.aliases.five_crop_transform(384,384))
output = tta_model(images,meta)