ttach----TTA模型多输入改写，包含image和meta

原创已于 2023-05-21 10:48:22 修改 · 624 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#ubuntu #linux #运维

于 2023-05-20 21:23:22 首次发布

动手学习深度学习专栏收录该内容

37 篇文章

订阅专栏

文章讲述了在使用ttach库进行图像处理时遇到的问题，即模型输入包含images和meta数据导致的错误。作者通过修改ttach/wrappers.py文件的前向传播函数解决了这个问题，但修改后模型的精度略有下降。此外，还提及了可自定义增强方法，如在aliases.py中添加five_crop_transform，以适应384x384的图像。最后，展示了修改后的代码运行示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在调用tta的时候由于我的模型输入有images和meta数据所以报错了，自己更改了下源码，能跑了但是精度低了一点点还有些问题。

1、修改wrappers

找到对应的文件miniconda3/lib/python3.8/site-packages/ttach/wrappers.py

接着按图修改前项传播函数

2、修改增强（选做）

找到文件miniconda3/lib/python3.8/site-packages/ttach/aliases.py，可以自己添加想要的增强

输出测试

tta_model = tta.ClassificationTTAWrapper(model, tta.aliases.five_crop_transform(384,384))
output = tta_model(images,meta)