Python Easyocr 图片文字识别

2022-07-27 08:33:16 浏览数 (1)

前段时间做了车牌识别相关的内容分享,参看:

车牌识别(1)-车牌数据集生成

车牌识别(2)-搭建车牌识别模型

今天给大家分享一个简单的OCR文本识别工具:easyocr。这个模块支持70多种语言的即用型OCR,包括中文,日文,韩文和泰文等。当然这个模块适当改进也可以用以车牌识别

1.安装

代码语言:javascript复制
pip install easyocr -i https://pypi.tuna.tsinghua.edu.cn/simple

它会安装除了模型文件之外的所有依赖,模型文件则会在运行代码的时候下载。

在pytorch网站上,请确保选择正确的CUDA版本。如果仅打算在CPU模式下运行,请选择CUDA = None。

我的电脑没安装gpu版本的pytorch,所以运行比较慢。

2.使用教程

代码语言:javascript复制
import easyocr
reader = easyocr.Reader(['ch_sim']) # ch_sim是Chinese simplified简写
result = reader.readtext('1.jpg')
print(result)
代码语言:javascript复制

运行的过程中会安装所需要的模型文件,像下面这样:

由于它的下载速度非常慢,而且经常会失败,因此建议先下载好模型文件,再将其放置到所需要的位置:

代码语言:javascript复制
文字检测模型(CRAFT)(必须)
https://pythondict.com/go/?url=https://github.com/JaidedAI/EasyOCR/releases/download/pre-v1.1.6/craft_mlt_25k.zip

中文(简体)模型(识别中文必须)
https://pythondict.com/go/?url=https://github.com/JaidedAI/EasyOCR/releases/download/pre-v1.1.6/chinese_sim.zip

中国(传统)模型
https://pythondict.com/go/?url=https://github.com/JaidedAI/EasyOCR/releases/download/pre-v1.1.6/chinese.zip

拉丁模型
https://pythondict.com/go/?url=https://github.com/JaidedAI/EasyOCR/releases/download/pre-v1.1.6/latin.zip

日本模型
https://pythondict.com/go/?url=https://github.com/JaidedAI/EasyOCR/releases/download/pre-v1.1.6/japanese.zip

韩文模型
https://pythondict.com/go/?url=https://github.com/JaidedAI/EasyOCR/releases/download/pre-v1.1.6/korean.zip

泰文模型
https://pythondict.com/go/?url=https://github.com/JaidedAI/EasyOCR/releases/download/pre-v1.1.6/thai.zip

阿拉伯文模型
https://pythondict.com/go/?url=https://github.com/JaidedAI/EasyOCR/releases/download/pre-v1.1.6/arabic.zip

如果下载速度太慢,请在公众号后台回复:easyocr(明天再试下载,今晚还没传输到网盘),下载文字检测模型(CRAFT)和中文简体模型文件包。

下载完模型后,将文件放到下面这个位置。

代码语言:javascript复制
Windows:C:Users用户名.EasyOCRmodel
Linux:~/ .EasyOCR / model

如下图所示:

重新执行脚本不会再提醒下载模型了,随便找的一个图片如下:

识别结果如下:

代码语言:javascript复制
[([[60, 308], [745, 308], [745, 447], [60, 447]], '文字识别提取', 0.9516711235046387), ([[77, 471], [725, 471], [725, 535], [77, 535]], '支持识别英法韩日俄德西葡语', 0.7867767214775085)]

输出采用列表格式,每个list分别表示对应文字的边界框、识别文本结果和置信度。

对于多语种的情况:

代码语言:javascript复制
import easyocr
reader = easyocr.Reader(['ch_sim', 'en'])
result = reader.readtext('chEN.jpg')
print(result)

效果如下:

总体效果还算不错,感兴趣的朋友可以试下车牌识别的效果,以及思考下如何改进车牌识别效果

0 人点赞