【OCR】总结Python中图片与pdf识别文本的方法

1、PaddleOCR

基于百度飞桨框架开发,模型丰富,支持多语言识别,包括中文、英文等。性能强大,适合复杂场景的文字识别

安装 PaddleOCR 库:

复制代码
pip install paddleocr

示例代码

复制代码
from paddleocr import PaddleOCR, draw_ocr
from PIL import Image

# 初始化 PaddleOCR
# 参数解释:
# `lang`:指定语言模型,如 'en'(英文)、'ch'(中文)等。
# `use_angle_cls`:是否启用文字方向分类器。
ocr = PaddleOCR(use_angle_cls=True, lang='en')  # 也可以设置为 'ch' 用于中文[^28^]

# 指定图片路径
img_path = 'example.jpg'  # 替换为你的图片路径

# 执行 OCR 识别
result = ocr.ocr(img_path, cls=True)  # `cls=True` 表示启用方向分类器

# 打印识别结果
for line in result:
    print(line)

# 可选:绘制识别结果并保存
if result:
    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, font_path='path/to/PaddleOCR/doc/fonts/simfang.ttf')
    im_show = Image.fromarray(im_show)
    im_show.save('result.jpg')  # 保存绘制后的图片[^28^]

2、RapidOCR

首先,确保安装了 RapidOCR 的 ONNXRuntime 版本,这是一个轻量级且高效的推理引擎:

复制代码
pip install rapidocr_onnxruntime

示例代码:识别数字和字母

以下代码展示了如何使用 RapidOCR 识别图片中的数字和字母,并仅打印识别结果:

复制代码
from rapidocr_onnxruntime import RapidOCR

# 初始化 OCR 引擎
ocr = RapidOCR()

# 指定图片路径
img_path = 'example.jpg'  # 替换为你的图片路径

# 执行识别
result, _ = ocr(img_path)

# 提取并打印识别结果(仅数字和字母)
if result:
    for line in result:
        text = line[1]  # 提取文字内容
        # 筛选只包含数字和字母的文本
        if text.isalnum():
            print(text)
else:
    print("未识别到文字")

注意事项

  • 图片路径 :确保 img_path 指向的图片包含数字或字母。
  • 语言设置:默认情况下,RapidOCR 支持中英文混合识别。如果需要识别其他语言,可以参考文档进行配置。
  • 环境要求:确保 Python 版本为 3.6 或更高。

3、EasyOCR

  • 特点:易于使用,支持多种语言(包括中文、英文等),基于深度学习技术,适合初学者和快速集成。

  • 安装方法

    复制代码
    pip install easyocr
  • 使用示例

    复制代码
    import easyocr
    
    reader = easyocr.Reader(['en', 'ch_sim'])  # 支持多语言
    img_path = 'example.jpg'
    result = reader.readtext(img_path)
    for line in result:
        print(line[1])  # 打印识别结果

4、Pytesseract

  • 特点:Tesseract 的 Python 封装,支持多种语言,使用简单,适合传统 OCR 任务。

  • 安装方法

    复制代码
    pip install pytesseract

    需要先安装 Tesseract OCR,可以从 Tesseract 官网 下载。

  • 使用示例

    复制代码
    from PIL import Image
    import pytesseract
    
    img_path = 'example.jpg'
    text = pytesseract.image_to_string(Image.open(img_path), lang='eng')
    print(text)  # 打印识别结果

5、DocTR

  • 特点:专注于文档分析和表格识别,能够提取文档中的结构化信息,适合处理复杂布局的文档。

  • 安装方法

    复制代码
    pip install python-doctr
  • 使用示例

    复制代码
    from doctr.models import ocr_predictor
    from doctr.io import DocumentFile
    
    img_path = 'example.jpg'
    doc = DocumentFile.from_images(img_path)
    model = ocr_predictor(pretrained=True)
    result = model(doc)
    for block in result.pages[0].blocks:
        for line in block.lines:
            for word in line.words:
                print(word.value)  # 打印识别结果

6、PyOCR

  • 特点:封装了多个 OCR 引擎(如 Tesseract、Cuneiform 等),提供了统一的接口。

  • 安装方法

    复制代码
    pip install pyocr
  • 使用示例

    复制代码
    import pyocr
    from PIL import Image
    
    tools = pyocr.get_available_tools()
    ocr_tool = tools[0]
    img_path = 'example.jpg'
    text = ocr_tool.image_to_string(Image.open(img_path), lang='eng')
    print(text)  # 打印识别结果

选择建议:

  • 速度优先 :推荐使用 RapidOCREasyOCR

  • 准确性优先 :推荐使用 PaddleOCR

  • 易用性优先 :推荐使用 EasyOCR

  • 文档分析优先 :推荐使用 docTR

注意:根据你的具体需求(如语言支持、应用场景、性能要求等),可以选择最适合的 OCR 库。

相关推荐
玩大数据的龙威24 分钟前
农经权二轮延包—各种地块示意图
python·arcgis
ZH154558913126 分钟前
Flutter for OpenHarmony Python学习助手实战:数据库操作与管理的实现
python·学习·flutter
belldeep35 分钟前
python:用 Flask 3 , mistune 2 和 mermaid.min.js 10.9 来实现 Markdown 中 mermaid 图表的渲染
javascript·python·flask
喵手36 分钟前
Python爬虫实战:电商价格监控系统 - 从定时任务到历史趋势分析的完整实战(附CSV导出 + SQLite持久化存储)!
爬虫·python·爬虫实战·零基础python爬虫教学·电商价格监控系统·从定时任务到历史趋势分析·采集结果sqlite存储
喵手1 小时前
Python爬虫实战:京东/淘宝搜索多页爬虫实战 - 从反爬对抗到数据入库的完整工程化方案(附CSV导出 + SQLite持久化存储)!
爬虫·python·爬虫实战·零基础python爬虫教学·京东淘宝页面数据采集·反爬对抗到数据入库·采集结果csv导出
B站_计算机毕业设计之家1 小时前
猫眼电影数据可视化与智能分析平台 | Python Flask框架 Echarts 推荐算法 爬虫 大数据 毕业设计源码
python·机器学习·信息可视化·flask·毕业设计·echarts·推荐算法
PPPPPaPeR.1 小时前
光学算法实战:深度解析镜片厚度对前后表面折射/反射的影响(纯Python实现)
开发语言·python·数码相机·算法
JaydenAI1 小时前
[拆解LangChain执行引擎] ManagedValue——一种特殊的只读虚拟通道
python·langchain
骇城迷影1 小时前
Makemore 核心面试题大汇总
人工智能·pytorch·python·深度学习·线性回归
长安牧笛1 小时前
反传统学习APP,摒弃固定课程顺序,根据用户做题正确性,学习速度,动态调整课程难度,比如某知识点学不会,自动推荐基础讲解和练习题,学习后再进阶,不搞一刀切。
python·编程语言