TTA(Test-Time Augmentation) 之Pytorch

TTA(Test-Time Augmentation) ,即测试时的数据增强

实现步骤如下:

将1个batch的数据通过flips, rotation, scale, etc.等操作生成batches
将各个batch分别输入网络
每个batch的masks/labels反向转换
通过mean, max, gmean, etc.合并各个batch预测的结果
最后输出最终的masks/labels

                   Input|           # input batch of images / / /|\ \ \      # apply augmentations (flips, rotation, scale, etc.)| | | | | | |     # pass augmented batches through model| | | | | | |     # reverse transformations for each batch of masks/labels\ \ \ / / /      # merge predictions (mean, max, gmean, etc.)|           # output batch of masks/labelsOutput

安装

        $ pip install ttach

使用方法如下

        import ttach as tta
...
model.load_state_dict(torch.load('models/%s/model.pth' %args.name)) 
model.eval()
tta_model = tta.SegmentationTTAWrapper(model, tta.aliases.d4_transform(), merge_mode='mean')
...

更多有关使用方法,可以看下面的参考链接

reference

https://github.com/qubvel/ttach

TTA(Test-Time Augmentation) 之Pytorch相关推荐

深度学习中的TTA(Test Time Augmentation)--测试时数据增强技术
定义:TTA(Test Time Augmentation):测试时数据增强方法:测试时将原始数据做不同形式的增强,然后取结果的平均值作为最终结果作用:可以进一步提升最终结果的精度原因:如果只对 ...
Pytorch快速搭建并训练CNN模型？
图像来自:快速上手笔记,PyTorch模型训练实用教程(附代码) - 知乎目录 1.数据处理模块搭建 2.模型构建 3.开始训练 4.评估模型 5.使用模型进行预测 6.保存模型 1.数据处理模块搭 ...
best single model of RSNA
对于[1]中的个单模型进行汇总: 用户模型数据集像素 LB得分备注 Tim Yee EfficientNet B1 224x224 0.098 Tim Yee EfficientNet B0 ...
论文推荐｜【KSII TIIS 2021】DP-LinkNet：一种用于古籍文档图像二值化的卷积网络(有源码)...
今日分享来自[KSII TIIS 2021]的论文『DP-LinkNet: A convolutional network for historical document image binariza ...
训练自己的数据_PyTorch版CenterNet训练自己的数据集
CenterNet(Objects as points)已经有一段时间了,之前这篇文章-[目标检测Anchor-Free]CVPR 2019 Object as Points(CenterNet)中讲 ...
Kaggle获奖者自述比赛求生指南：我们如何“穿越”亚马逊热带雨林
本文来自AI新媒体量子位(QbitAI) 大家好,我是思聪 · 格里尔斯,我将向您展示如何从世界上某些竞争最激烈的比赛中拿到金牌.我将面临一个月的比赛挑战,在这些比赛中缺乏正确的求生技巧,你甚至拿不到 ...
PaperWeekly第44期 | Kaggle求生
大家好,我是思聪 · 格里尔斯,我将向您展示如何从世界上某些竞争最激烈的比赛中拿到金牌.我将面临一个月的比赛挑战,在这些比赛中缺乏正确的求生技巧,你甚至拿不到铜牌.这次,我来到了亚马逊热带雨林. 当我 ...
基于YOLOV5动态检测19种类别
#1 makeTxt.py(将训练数据自动划分为训练集.验证集和测试集)... 2 #2 voc_label.py(将VOC格式数据集转换成yolo数据集)... 3 #3 tube.yaml 6 # ...
【图像分割】卫星遥感影像道路分割：D-LinkNet算法解读
前言因为毕设中的部分内容涉及到卫星遥感影像道路分割,因此去对相关算法做了一些调研. 本文所使用数据集为DeepGlobe,来自于CVPR2018年的一个挑战赛:DeepGlobe Road Extr ...

TTA(Test-Time Augmentation) 之Pytorch

reference

TTA(Test-Time Augmentation) 之Pytorch相关推荐

最新文章

热门文章