跳转至

调优尝试教程

Note

以下尝试情况，均已经在在线demo中实现，看完之后，可移步那里去尝试。

引言

  • 由于小伙伴们使用OCR的场景多种多样，单一的参数配置往往不能满足要求，这就涉及到基于现有模型，通过调节参数来达到正确识别的目的。
  • 基本的原则是尽量不去动模型内部，降低使用成本。
  • 本篇文章就来说说在识别效果不好时，如何去调优。
  • 因此博客为一家之言，难免会挂一漏万，希望小伙伴多多指出，共同维护这份教程。

常见错例种类

本地使用过程中，识别结果：

  • 个别字丢失
  • 个别字识别错误

个别字丢失调优篇

  1. 查看图像是否shape太小，导致文本检测丢失。

    • 对于长度较长，高度较小的图像，可尝试对该图像高度做上下补充，来减小文字在原始图像中比例，利于文本检测。类似下图这种，左侧一条为原始图像，右侧为上下补边的图像。 Snipaste_2023-04-09_14-01-45
    • padding代码参考：
     1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
    from typing import Tuple

import cv2
import numpy as np


def padding_img(img: np.ndarray,
                padding_value: Tuple[int, int, int, int],
                padding_color: Tuple = (0, 0, 0)) -> np.ndarray:
    padded_img = cv2.copyMakeBorder(img,
                                    padding_value[0],
                                    padding_value[1],
                                    padding_value[2],
                                    padding_value[3],
                                    cv2.BORDER_CONSTANT,
                                    value=padding_color)
    return padded_img

img = cv2.imread('xxx.jpg')

# padding_value: (top, bottom, left, right)
# 对该图像上下各补充10像素的白边
padded_img = padding_img(img, (10, 10, 0, 0), (255, 255, 255))

  2. 将padding之后的图像再次送入RapidOCR中尝试，查看是否仍然出现上述问题。

  3. 将该图像上传到在线demo中，查看能否复现问题？

  4. 如不能复现，着重比对环境是否与在线demo中相关包使用版本不一致情况？

  5. 在线demo所用的各个包版本情况：

    Python: 3.8
onnxruntime: 1.14.1
rapidocr_onnxruntime: 1.2.5

  6. 如能复现，尝试调节在线demo中的三个参数，注意控制变量调节参数

  7. 首先调节box_thresh参数，该参数用来控制文本检测部分检测所得框是文本的概率。个别字丢失，一般是单独的字丢失，很大可能是文本检测模型没有检测到该独立的文字。尝试调低box_thresh值，查看是否可以检出丢失的文字。（没有找到合适的例子）
  8. 固定box_thresh，调节text_score参数。该参数是控制识别文本结果正确的概率。尝试调低text_score值，查看是否可以识别出丢失的文字。
  9. 固定box_thresh和text_score两个参数，调节unclip_ratio参数。该参数用来控制文本检测所得文本框的大小。尝试调大unclip_ratio值，查看是否可以识别出丢失的文字。
  10. 如果调节三个参数也不能解决问题，只能尝试server版的文本检测和文本识别模型了。
  11. 首先更换server版的文本检测模型，查看是否解决问题
  12. 在第1步基础上，更换server版的文本识别模型，查看是否解决问题
  13. 尝试更换不同版本模型，来查看效果。具体可在在线demo中尝试

个别字识别错误调优篇

个别字错误的情况，例如：成识别为戍，这种一般是文本识别模型的问题。

情况一：轻量中英文模型识别对个别汉字识别错误

  1. 尝试padding图像，再重新识别
  2. 更换不同版本的识别模型，包括v2、v3和server版尝试

情况二：轻量中英文模型对个别英文或数字识别错误

  1. 更换不同版本的识别模型，包括v2、v3和server版尝试
  2. 如果是只识别英文单词和数字，可以更换英文和数字专有模型（en_PP-OCRv3_rec_infer.onnx和en_number_mobile_v2.0_rec_infer.onnx）

评论