keras模型的测试时间图像增强。
tta-wrapper的Python项目详细描述
tta包装器
测试keras图像分割和分类模型的时间分析包装器。
wrapper将增强层添加到您的路缘石模型中,如下所示:
````
input
移位)
你的Keras模型
反向变换
//合并预测(平均值,最大值,gmean)
输出掩码;形状1,h,w,c
输出
``````
参数
-`h翻转`-bool,水平翻转增大
-`v懔flip`-bool,垂直翻转增大
-`rotataion`-列表,允许角度-90、180、270
-`h懔shift`-int列表,像素水平移位增大
-`v懔shift`-int列表,像素垂直移位增大
-`add`-int/float列表,加法因子(aug懔图像=图像+因子)
-`mul`-int/float列表,加法因子(aug_image=image*factor)
-`contrast`-int/float列表,对比度调整因子(aug_image=(image-mean)*factor+mean)
-`merge`-mean之一,“gmean”和“max”-合并扩展预测的模式
tta_u包装导入tta_u分割
model=load_u模型('path/to/model.h5')
tta_u模型=tta_u分割(model,h_u flip=true,rotation_angles=(90,270),
h_u shifts=(-5,5),merge='mean')
y=tta_模型.预测(x)
``````
标签: