Bootstrap

利用Paddle OCR进行文字识别

利用Paddle OCR HubServing + Java Demo进行文字识别

需求描述

需要批量过滤一批本地图片,将含有网址水印的图片剔除

实现逻辑

使用python开源库 chinese_ocr_db_crnn_server 部署成本地http服务,使用Java进行http调用,获取图片中的文字信息,进行文字比较。

经测试,识别效果很好,准确率也很高,至于性能需要看服务器的配置了。

运行环境

名称 版本
操作系统 Centos7.9
python 3.7.0
pip3 22.1.1
paddlepaddle 2.3

1. 安装paddlepaddle

python -m pip install paddlepaddle==2.3.0 -i https://mirror.baidu.com/pypi/simple
## 验证安装是否成功,执行下面命令出现 PaddlePaddle is installed successfully 即是安装成功
python3
import paddle
paddle.utils.run_check()

安装的过程中由于硬件和操作系统环境不同,可能会出现某些依赖缺少的情况,需要自己进行排错安装

2. 下载Paddle OCR的源码

## git仓库克隆源码
git clone https://gitee.com/PaddlePaddle/PaddleOCR
## 移动源码到你想要的位置
mv ./PaddleOCR /PaddleOCR
##进入PaddleOCR目录进行依赖的安装
cd /PaddleOCR
pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

3. 模型库下载

需要下载3个模型,分别是检测模型,方向分类器,识别模型

模型 链接
检测模型 https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_det_infer.tar
方向分类器 https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_cls_infer.tar
识别模型 https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_rec_infer.tar

下载之后上传到PaddleOCR/interface目录下,interface如果无此目录则使用mkdir新建,然后解压

cd /PaddleOCR
mkdir interface
## 模型解压
tar -xvf ch_ppocr_mobile_v2.0_cls_infer.tar
tar -xvf ch_ppocr_server_v2.0_det_infer.tar
tar -xvf ch_ppocr_server_v2.0_rec_infer.tar

4. 安装HubServing服务

cd /PaddleOCR
export PYTHONPATH=.
## 下载服务模型
hub install chinese_ocr_db_crnn_server==1.1.1
## 服务启动
nohup hub serving start -m chinese_ocr_db_crnn_server >> /PaddleOCR/hub.log  2>&1 &
##出现 Running on http:/127.0.0.1:8866之类的信息表示启动成功
## 服务停止命令
kill -9 pid
## 或者
hub serving stop

5. 编写Java单元测试

package com.example.test.util;

import java.io.*;
import java.util.ArrayList;
import java.util.List;

/**
 * @autor Hou Dehong
 * @date 2022/5/19
 * @description
 */
public class PaddleDemo2 {
   

    public static List<String> list = new ArrayList<>();
    public static void main(String[] args) throws Exception {
   
        String url = "http://10.10.102.208:8866/predict/chinese_ocr_db_crnn_server";
        String dirOri = "C:\\Users\\Administrator\\Desktop\\";
        String fileName = "test.jpg";
        // 读取文件
        byte[] imgData = readFileByBytes(dirOri + fileName);
        String imgStr = Base64Util.encode(imgData);
        String param = "{\"images\":[\"" + imgStr + "\"]}";
        System.out.println(fileName);
        String result = HttpUtil.postGeneralUrl(url, "application/json", param,"UTF-8");
        System.out.println(result);
    }

    public static byte[] readFileByBytes(String filePath) throws IOException {
   
        File file = new File(filePath);
        if (!file.exists()) {
   
            throw new FileNotFoundException(filePath);
        } else {
   
            ByteArrayOutputStream bos = new ByteArrayOutputStream((int) file
;