IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 人工智能 -> 【python OCR】PaddleOCR简单demo入门 -> 正文阅读

[人工智能]【python OCR】PaddleOCR简单demo入门

作者:recommend-item-box type_blog clearfix

PaddleOCR开源地址:

https://github.com/PaddlePaddle/PaddleOCR

参考资料:快速入门PaddleOCR,并使用其开发一个搜题小工具

一、配置环境

安装依赖库:CPU版本的paddlepaddle

pip install paddlepaddle

安装PaddleOCR

pip install “paddleocr>=2.0.1”

二、下载预训练模型

PaddleOCR 分为 Detection(文本检测)、 Direction classifier(方向分类器)和Recognition(文本识别)三部分,因此需要三个模型。官方代码仓中有模型下载地址:

在这里插入图片描述
其中文本检测有三个模型,分别是MobileNetV3、ResNet18_vd和ResNet50,其中最常使用的是MobileNetV3模型,整体比较小,适合应用于手机端。文本识别只有一个MobileNetV3预训练模型。方向分类器使用默认的模型。

把最后一行的通用检测模型(143.4M)的三个模型(tar压缩包)下载下来并解压,PaddleOCR加载的model是解压后的文件夹,而不是压缩包。

三、简单代码入门

随便找一张图片进行OCR文本检测:

在这里插入图片描述

3.1 代码:

参考官方的demo:demo

from paddleocr import PaddleOCR, draw_ocr
from PIL import Image


# load model
# Paddleocr目前支持中英文、英文、法语、德语、韩语、日语,可以通过修改 lang参数进行切换
# lang参数依次为`ch`, `en`, `french`, `german`, `korean`, `japan`
ocr = PaddleOCR(lang="ch",
                use_gpu=False,
                det_model_dir="../paddleORC_model/ch_ppocr_server_v2.0_det_infer/",
                cls_model_dir="ch_ppocr_mobile_v2.0_cls_infer/",
                rec_model_dir="ch_ppocr_server_v2.0_rec_infer/")

# load dataset
img_path = 'test.jpg'
resul = ocr.ocr(img_path)
for line in result:
    print(line)

# 注:
# result是一个list,每个item包含了文本框,文字和识别置信度
# line的格式为:
# [[[3.0, 149.0], [43.0, 149.0], [43.0, 163.0], [3.0, 163.0]], ('人心安', 0.6762619018554688)]
# 文字框 boxes = line[0],包含文字框的四个角的(x,y)坐标
# 文字 txts = line[1][0]
# 识别置信度 scores = line[1][1]

# visual
image = Image.open(img_path).convert('RGB')
boxes = [line[0] for line in result]
txts = [line[1][0] for line in result]
scores = [line[1][1] for line in result]
im_show = draw_ocr(image, boxes, txts, scores)
im_show = Image.fromarray(im_show)
im_show.save('result.jpg')

3.2 识别结果

在这里插入图片描述
检测的输出结果:

[[[47.0, 35.0], [306.0, 35.0], [306.0, 90.0], [47.0, 90.0]], ('免夹免拉', 0.989071249961853)]
[[[52.0, 109.0], [276.0, 109.0], [276.0, 130.0], [52.0, 130.0]], ('操作简单居家可做', 0.988811731338501)]
[[[52.0, 146.0], [277.0, 146.0], [277.0, 167.0], [52.0, 167.0]], ('温和配方健康直发', 0.9917734861373901)]
[[[140.0, 441.0], [186.0, 441.0], [186.0, 455.0], [140.0, 455.0]], ('STRAIT', 0.9612524509429932)]
[[[113.0, 464.0], [208.0, 453.0], [213.0, 494.0], [118.0, 505.0]], ('glatt', 0.9871892929077148)]
[[[31.0, 561.0], [52.0, 560.0], [56.0, 642.0], [35.0, 643.0]], ('219PLUS', 0.7543826103210449)]
[[[609.0, 586.0], [774.0, 586.0], [774.0, 631.0], [609.0, 631.0]], ('活动价:', 0.9964019060134888)]
[[[594.0, 639.0], [799.0, 645.0], [794.0, 798.0], [589.0, 792.0]], ('45', 0.9980571269989014)]
[[[581.0, 652.0], [603.0, 652.0], [603.0, 680.0], [581.0, 680.0]], ('?', 0.8542420268058777)]
[[[143.0, 680.0], [473.0, 680.0], [473.0, 711.0], [143.0, 711.0]], ('下单赠送全套工具', 0.9827134609222412)]
[[[12.0, 749.0], [545.0, 747.0], [545.0, 781.0], [12.0, 783.0]], ('一梳就直免拉免夹家用直发膏', 0.9552955627441406)]

可以看到识别结果还是非常理想的,尽管模型中选择的语言 lang=“ch” ,但中英文都可以识别出来。

3.3 关于识别结果的说明

result = ocr.ocr(img_path)
for line in result:

result是一个list,每个item包含了文本框,文字和识别置信度
line的格式为:
[[[3.0, 149.0], [43.0, 149.0], [43.0, 163.0], [3.0, 163.0]], ('人心安', 0.6762619018554688)]
文字框 boxes = line[0],包含文字框的四个角的(x,y)坐标
文字 txts = line[1][0]
识别置信度 scores = line[1][1]

paddleOCR内置的可视化函数 draw_ocr() 方法需要 image, boxes, txts, scores 四个参数就可以实现可视化。当然该函数还可以接受其他参数,如字体文件路径。

  人工智能 最新文章
2022吴恩达机器学习课程——第二课(神经网
第十五章 规则学习
FixMatch: Simplifying Semi-Supervised Le
数据挖掘Java——Kmeans算法的实现
大脑皮层的分割方法
【翻译】GPT-3是如何工作的
论文笔记:TEACHTEXT: CrossModal Generaliz
python从零学(六)
详解Python 3.x 导入(import)
【答读者问27】backtrader不支持最新版本的
上一篇文章      下一篇文章      查看所有文章
加:2022-05-10 11:54:11  更:2022-05-10 11:55:10 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/26 7:31:34-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码