paddle ocr开启mkldnn加速后图片识别精度下降

背景

使用ocr_system调用微调模型时发现部分图片在设置enable_mkldnn=True后出现识别精度下降问题

环境:

paddle==2.5.1
paddlehub==2.3.1
python==3.8.6

解决方案

tools\infer\utility.py文件中 line 269行左右,
在代码config.delete_pass("matmul_transpose_reshape_fuse_pass")
下面添加两行代码:
config.delete_pass("fc_mkldnn_pass")
config.delete_pass("fc_act_mkldnn_fuse_pass

参考:

https://github.com/PaddlePaddle/PaddleOCR/issues/10497

官方说会在后续版本中处理,但是截止发文时间点,paddle==2.5.2版本中仍无法解决该问题

你可能感兴趣的:(图像识别,paddle,ocr)