pdocr-rpc 1.0.0

PaddleOCR-RPC

RPC-PaddleOCR

基于 PaddleOCR 部署的 RPC 服务。

提供了一个简单易用的函数 ocr，通过不同的参数控制返回不同的值。

安装

pip install pdocr-rpc

1、客户端

1.1、使用方法

# ocr.py

def ocr(*target_strings, picture_abspath=None, similarity=0.6, return_default=False, return_first=False, lang="ch"):
    """
     通过 OCR 进行识别。
    :param target_strings: 
        目标字符,识别一个字符串或多个字符串,并返回其在图片中的坐标;
        如果不传参，返回图片中识别到的所有字符串。
    :param picture_abspath: 要识别的图片路径。
    :param similarity: 匹配度。
    :param return_default: 返回识别的原生数据。
    :param return_first: 只返回第一个,默认为 False,返回识别到的所有数据。
    :param lang: `ch`, `en`, `fr`, `german`, `korean`, `japan`
    :return: 返回的坐标是目标字符串所在行的中心坐标。
    """

1.2、使用场景

1.1.1、指定某张图片识别

ocr("~/Desktop/test.png")

返回识别图片 test.png 的内容。

1.1.2、识别当前屏幕

ocr()

自动识别当前整个屏幕的所有内容。

1.1.3、指定查找某个字符串的坐标

ocr("天天向上")

返回当前屏幕中，“天天向上”的坐标，如果存在多个，则返回一个字典。

通过不同的参数可以改变返回值的类型；

1.3、安装依赖

客户端仅需要安装截图工具；

• Windows 上使用：

pip install pillow

• Linux 上使用：

PIL ImageGrab 模块在部分的 Linux 上可能存在问题，报错：ImportError: ImageGrab is macOS and Windows only ；

Linux 上推荐安装 pyscreenshot；

sudo pip3 install pyscreenshot

然后修改导入代码：

# Linux
import pyscreenshot as ImageGrab
# Windows or macOS
# from PIL import ImageGrab

2、服务端

使用 Linux 操作系统进行部署，debian、ubuntu、centos、UOS、deepin 等常见的发行版都是可以的。

推荐 pipenv 进行环境搭建；

安装 pipenv ：

sudo pip3 install pipenv

新建一个目录作为环境包 ocr_env：

cd ~
mkdir ocr_env

创建 python 3.7 环境：

cd ocr_env
pipenv --python 3.7

安装 OCR 依赖包：

pipenv install paddlepaddle -i https://mirror.baidu.com/pypi/simple
pipenv install "paddleocr>=2.0.1" -i https://mirror.baidu.com/pypi/simple

不出意外，这样就把依赖安装好了。

2.1、启动服务

将 ocr_server.py 文件拷贝到 ocr_env 目录，后台执行它就好了：

cd ocr_env
nohup pipenv run python ocr_server.py &

2.2、配置开机自启

你肯定不想每次机器重启之后都需要手动启动服务，因此我们需要配置开机自启。

写开机自启服务文件：

sudo vim /lib/systemd/system/ocr.service

autoocr 名称你可以自定义，写入以下内容：

[Unit]
Description=OCR Service
After=multi-user.target

[Service]
User=uos
Group=uos
Type=idle
WorkingDirectory=/home/uos/ocr_env
ExecStart=pipenv run python ocr_server.py

[Install]
WantedBy=multi-user.target

注意替换你的${USER}

修改配置文件的权限：

sudo chmod 644 /lib/systemd/system/ocr.service

自启服务生效：

sudo systemctl daemon-reload
sudo systemctl enable ocr.service

查看服务状态：

sudo systemctl status ocr.service

你可以再重启下电脑，看看服务是不是正常启动了，没报错就 OK 了。

2.3、缓存

在 ocr_env/pic 目录下保存了识别的一些缓存图片文件，您可能需要定期进行删除；

当然，你可以使用定时任务对缓存文件进行清理，例如 crontab、Jenkins 任务等。