如何才能从图片中识别出文字呢，今天我们将使用 飞桨的 PaddleOCR 进行图片文字识别。

官网地址: https://www.paddlepaddle.org.cn/hub/scene/ocr

学到什么

• python 图片文字提取
• 文件目录或文件更新监控

主题:

使用 PaddleOCR 提取图片中的文字，图片由其他程序进行下载或者人工保存到文件夹，使用 watchfiles 进行文件目录，检测新保存的文件进行文字提取

关键词:

• Python 图片文字识别
• Python 监控文件和目录变化

安装相关库

pip3 install paddlepaddle paddleocr
pip3 install watchfiles

使用 OCR 提取图片中的文字

ocr = PaddleOCR(use_angle_cls=True, lang='ch')


def image_ocr(path):
    """ 图片文字识别
    :param path: 图片路径
    :return: List[str]
    """
    result = ocr.ocr(str(path), det=True)
    return result

上面我们对图片中的中文进行提取结果

文件监控部分

使用watchfiles监控文件变更 主要是watch 函数

• path 监控目录路径
• rust_timeout 超时时间 超时后返回空changes
• watch_filter 自定义文件过滤

def watch_image(*path: str) -> Generator[Tuple[Change, str], None, None]:
    """文件监控
    :param path: 监控目录 可输入多个目录
    :return:
    """
    for changes in watch(*path, debounce=1000, rust_timeout=300_000, watch_filter=ImageFilter()):
        for event, change_file in changes:
            logger.info(f'{event.name} {change_file}')
            yield event, change_file

最后贴出出完整代码 如需定制或者制作应用程序(可提供)

# _*_coding: utf-8 _*_
# @Time    : 2019/12/18 15:54
# @Author  : Taihui

import pathlib
import pprint
from typing import Generator, Tuple

from paddleocr import PaddleOCR
from loguru import logger
from watchfiles import watch, Change, DefaultFilter


ocr = PaddleOCR(use_angle_cls=True, lang='ch')


def image_ocr(path):
    """ 图片文字识别
    :param path: 图片路径
    :return: List[str]
    """
    result = ocr.ocr(str(path), det=True)
    return result


class ImageFilter(DefaultFilter):
    allowed_extensions = ['.png', '.jpg', '.jpeg', '.gif']

    def __call__(self, event: Change, path: str) -> bool:
        """ 过滤无效扩展名
        :param event: 文件变更事件 add、modify、delete
        :param path: 文件路径
        :return: bool
        """
        if not pathlib.Path(path).suffix in self.allowed_extensions:
            return False
        return super().__call__(event, path)


def watch_image(*path: str) -> Generator[Tuple[Change, str], None, None]:
    """文件监控
    :param path: 监控目录 可输入多个目录
    :return:
    """
    for changes in watch(*path, debounce=1000, rust_timeout=300_000, watch_filter=ImageFilter()):
        for event, change_file in changes:
            logger.info(f'{event.name} {change_file}')
            yield event, change_file


def main(*path):
    for event, change_file in watch_image(*path):
        if event == Change.added or event == Change.modified:
            try:
                pprint.pp(image_ocr(change_file))
            except FileNotFoundError:
                pass


if __name__ == '__main__':
    main('images/')