tesseract5.0 图片训练模型实战

最近接了个活,有大批的图片需要文字识别

图片参考

 

是印刷字体打印后再扫描的图片,看到这种任务,首先想到的是用tesseract进行识别,印刷字体识别率很高

拿出工具进行识别分析,效果感人,5段文字,有3段都有识别错的,2和Z,3和S

进过多个图片测试,基本不可用,识别错的太多了

 

 

 对接百度通用识别,使用高精度版,准确率基本100%,不过百度账户有数量限制,客户有几千上万的图像需要识别,这个方式行不通,毕竟能免费为啥要付钱呢

免费测试网址:https://cloud.baidu.com/product/ocr/general

通过资料查阅,发现可以对某一类图片进行训练,即可大幅提升识别准确率

训练过程参考:

https://www.cnblogs.com/cnlian/p/5765871.html

虽然这个教程是3.0的,但是5.0一样训练,不影响流程

通过选取40个图片为训练样本,合并成一个tif,生成box文件,然后使用jTessBoxEditor修正错误,然后训练出模型文件

使用新的模型文件进行识别,果然识别率大幅上升,测试准确度有99.9%,大批量测试速度也很快,1秒100张的速度

 

 

 ==================================================================================

 ==================================================================================

这边分享一些中间遇到的坑和经验:

1.32位模型和64位模型不通用,而且32位训练我这边出问题直接训练不成功

2.提示无法读取box:APPLY_BOXES: boxfile line 3/G ((80,0),(80,0)): FAILURE! Couldn't find a matching blob

  这种错误大概率是训练图片太脏引起的,

  如下左边这种图片就不行,需要使用opencv进行二值化处理,白底黑字,参照右图

 

 

3.如果百度识别效果还可以,建议使用百度识别修改文件名,这样大幅减少人工识别纠正的工作量

4.建议使用opencv进行处理,寻找轮廓,自己生成box文件,读取文件名来匹配每个轮廓的文字内容,这样可以极大的减少工作量

  (jTessBoxEditor太难用了, 坐标得手动改,文字也要一个个改,不能放大,你要是改个100个,估计得崩溃)

  

  box文件,格式为【文字内容,p1, p2, p3, p4, 图片序号】

   

#opencv的rect转换为box文件格式坐标
def get_tesseract_box(x, y, w, h, height):
    p1 = x
    p4 = height - y
    p2 = p4 - h
    p3 = x + w
    return [p1, p2, p3, p4]

 height为图像高度

5.如果需要训练的样本比较多,建议首先使用小批量的图片训练出一个模型,然后使用这个模型来识别,作为原始图片名,可以大幅减少工作量

 

posted @ 2021-10-24 15:31  darkspr  阅读(2004)  评论(0编辑  收藏  举报