ODAch是一个用于pytorch 2d目标探测器的测试时间增强工具。
odach的Python项目详细描述
ODAch，Pytorch的目标检测TTA工具
ODA是一种用于二维目标探测器的测试时间增强（TTA）工具。在
用于Kaggle目标探测比赛。在
：星：如果它对你有帮助的话！；）
安装
pip install odach
使用
请参见Example.ipynb。在
设置非常简单，类似于ttach。在
单尺度TTA
importodachasoda# Declare TTA variationstta=[oda.HorizontalFlip(),oda.VerticalFlip(),oda.Rotate90(),oda.Multiply(0.9),oda.Multiply(1.1)]# load imageimg=loadimg(impath)# wrap model and ttatta_model=oda.TTAWrapper(model,tta)# Execute TTA!boxes,scores,labels=tta_model(img)
多尺度TTA
^{pr2}$- 在
这些框也由nms过滤（wbf默认）。在
在 - 在
图像大小应为方形。在
在
模型输出包装
- 在
包装您的检测模型，以便输出类似于torchvision frcnn格式： [[“框”：[[x，y，x2，y2]，[]，…]，“标签”：[0,1，…]，“分数”：[1.0，0.8，…]]
在 - 在
效率设置示例 https://www.kaggle.com/kyoshioka47/example-of-2d-single-scale-tta-with-odach/
在
# wrap effdetoda_effdet=oda.wrap_effdet(effdet)# Declare TTA variationstta=[oda.HorizontalFlip(),oda.VerticalFlip(),oda.Rotate90()]# Declare scales to ttascale=[1]# wrap model and ttatta_model=oda.TTAWrapper(oda_effdet,tta,scale)
示例
全球小麦检测
谢谢
nms、wbf来自https://kaggle.com/zfturbo
tta基于https://github.com/qubvel/ttach、https://github.com/andrewekhalel/edafa/tree/master/edafa和{a7}
- 项目
标签：
欢迎加入QQ群-->： 979659372
