Skip to content

Latest commit

 

History

History
25 lines (13 loc) · 930 Bytes

README.md

File metadata and controls

25 lines (13 loc) · 930 Bytes

DBnet_TensorRT

参考的原始版本

模型训练代码

wts模型地址-百度网盘

提取码:zzl3

icdar2015训练的pth模型文件

提取码:spv2

修改

1.训练代码将上采样层的最邻近插值(nearest)修改为双线性插值(bilinear)

2.去掉了减均值除方差前处理(训练也去掉),感觉没用,主要是会导致trt的前处理耗时,至少占上面代码的整个流程的40%,取决于输入图片大小

3.原始tensorrt代码后处理没有分数,box扩大代码有bug,不能按照长宽进行等比例缩放,导致box不准

4.处理性能提高,640x640输入,整个流程耗时17ms

5.训练的模型,recall:90%,precision:77%,hmean(f1):82.7%,resnet18+fpn