devbox
我家小花儿
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

paddle-OCRv2预测部署_ch_pp-ocrv2_det_slim_quant_infer

作者：我家小花儿 | 2024-04-05 07:34:12

赞

踩

ch_pp-ocrv2_det_slim_quant_infer

要点：

  • 奇奇怪怪

1 预测部署简介与总览

本章主要介绍PP-OCRv2系统的高性能推理方法、服务化部署方法以及端侧部署方法。通过本章的学习，您可以学习到：

  • 根据不同的场景挑选合适的预测部署方法
  • PP-OCRv2系列模型在不同场景下的推理方法
  • Paddle Inference、Paddle Serving、Paddle Lite的推理部署开发方法

1.1 简介

在前面几个章节中，我们通过模型训练的方法，得到了训练好的模型，在使用它去预测的时候，我们首先需要定义好模型，然后加载训练好的模型，再将预处理之后的数据送进网络中进行预测、后处理，得到最终的结果。使用这种方法去进行预测，调试方便，但是预测效率比较低下。

针对上面的问题，对于训练得到的模型，一般有下面2种离线预测的方式。

  1. 基于训练引擎推理，也就是和训练使用同一套引擎，它调试方便，便于我们迅速定位问题，验证正确性。多为Python语言。
  2. 基于预测引擎推理，将训练得到的模型进行转换，去掉与推理无关的部分，使用这种方法可以加速预测。多为Python语言或C++。

具体地，PaddleOCR 针对不同应用场景，提供了三种预测部署方案。

  • Inference的离线预测，这种方式主要应用在对预测响应的及时性要求不高，特别是需要大量图片预测的场景。像文档电子化、广告信息提取等。虽然不能及时响应预测请求，但是没有网络延时，计算效率比较高，数据安全性很高。
  • Serving服务化部署，这种方式主要应用在对预测响应的及时性要求很高的场景，像商业化OCR的API接口、实时拍照翻译、拍题等场景。虽然这种方式能及时对预测需求及时响应，但是网络耗时开销比较大，GPU利用率往往不高，而且存在数据安全风险。
  • Lite端侧部署，这种方式主要希望模型部署到手机、机器人等端侧设备上，主要考虑部署方便和数据安全性，像手机APP里面身份证识别、银行卡识别，工业应用场景的仪表监控识别等。这种方法，对OCR模型的大小比较敏感。虽然没有网络延时，数据安全性很高，但是由于算力限制，预测效率不高。

本章基于PP-OCRv2，介绍文本检测、识别以及系统串联预测推理与部署过程。

1.2 环境准备

体验本章节内容需要首先下载PaddleOCR代码，安装相关依赖，具体命令如下

  1. import os
  2.  
  3. os.chdir("/home/aistudio")
  4. # 下载代码
  5. !git clone https://gitee.com/paddlepaddle/PaddleOCR.git
  6. os.chdir("/home/aistudio/PaddleOCR")
  7. # 安装运行所需要的whl包
  8. !pip install -U pip
  9. !pip install -r requirements.txt
  10. # VQA任务中需要用到该库
  11. !pip install paddlenlp==2.2.1
  12.  
  13. # 导入一些库
  14. import cv2
  15. import matplotlib.pyplot as plt
  16. %matplotlib inline
  17. import numpy as np
  18. import os

2 基于Paddle Inference的python推理

2.1 简介

在项目中，模型的推理性能直接影响项目成本，因此我们期望一个训练好的模型的模型可以拥有更快的推理速度。直接基于训练引擎进行预测，模型中包含与训练相关的算子，因此效率一般较低；而且需要定义模型，难以与训练代码解耦。Paddle Inference应运而生。它是飞桨的原生推理库，作用于服务器端和云端，提供高性能的推理能力。由于能力直接基于飞桨的训练算子，因此Paddle Inference 可以通用支持飞桨训练出的所有模型。

考虑到大家的使用场景差异很大，Paddle Inference针对不同平台不同的应用场景进行了深度的适配优化，做到高吞吐、低时延，保证了飞桨模型在服务器端即训即用，快速部署。

本章主要介绍基于Paddle Inference的PP-OCRv2预测推理过程，更多关于Paddle Inference的介绍可以参考：Paddle Inference 介绍。

在基于Paddle Inference进行模型推理时，一般有以下几个步骤。

PP-OCRv2系统包含文字检测、方向分类器和文字识别3个模型，下面分别介绍这3个模型基于Paddle Inference的推理过程。

2.2 PP-OCRv2文字检测模型推理

PaddleOCR中，在基于文字检测模型进行推理时，需要通过参数image_dir指定单张图像或者图像集合的路径、参数det_model_dir, 指定检测的 inference 模型路径。

下面进行最新的超轻量文本检测模型推理实战，更多的模型和使用方法请参考文本检测预测教程。

更多等算法超参数的介绍可以参考PaddleOCR Inference推理相关参数介绍。

2.2.1 准备数据和环境

在最开始已经安装好了Paddle以及相应的依赖，这里环境已经准备好了。

检测示例数据在doc/imgs文件夹下面，部分数据如下所示。

  1. # 切换目录
  2. os.chdir("/home/aistudio/PaddleOCR")
  3.  
  4. # 查看数据
  5. !ls doc/imgs/
  6.  
  7. # 选择2张图像可视化
  8. img1 = cv2.imread("doc/imgs/00006737.jpg")
  9. img2 = cv2.imread("doc/imgs/00056221.jpg")
  10. plt.figure(figsize=(15, 6))
  11. plt.subplot(1, 2, 1)
  12. plt.imshow(img1[:,:,::-1])
  13. plt.subplot(1, 2, 2)
  14. plt.imshow(img2[:,:,::-1])
  15. plt.show()

2.2.2 准备推理模型

下载推理模型并解压，放在inference目录下面。

  1. # 下载模型
  2. !mkdir inference
  3. !cd inference && wget https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_det_infer.tar -O ch_PP-OCRv2_det_infer.tar && tar -xf ch_PP-OCRv2_det_infer.tar
  4. !tree -h inference/ch_PP-OCRv2_det_infer
  • 如果您希望导出自己训练得到的模型，使用Paddle Inference部署，那么可以使用下面的命令将预训练模型使用动转静的方法，转化为推理模型。
  1. # 参考代码
  2. # https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.4/tools/export_model.py
  3. # 下载预训练模型
  4. !wget https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_det_distill_train.tar && tar -xf ch_PP-OCRv2_det_distill_train.tar && rm ch_PP-OCRv2_det_distill_train.tar
  5. # 导出推理模型
  6. !python tools/export_model.py -c configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_cml.yml \
  7.     -o Global.pretrained_model="ch_PP-OCRv2_det_distill_train/best_accuracy" \
  8.     Global.save_inference_dir="./my_model"
  9. # PP-OCRv2检测模型包含3个子网络：教师、学生、学生2，因此导出时，包含3个子文件，实际推理时，使用其中1个学生网络进行推理即可
  10. !tree -h my_model

2.2.3 文本检测功能初探

我们先来看看加载加载推理模型预测得到的结果。

  1. # 参考代码
  2. # https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.4/tools/infer/predict_det.py
  3. # 预测
  4. !python tools/infer/predict_det.py --image_dir="./doc/imgs/00018069.jpg" --det_model_dir="./inference/ch_PP-OCRv2_det_infer" --use_gpu=False
  5.  
  6. # 读取图像并显示出来，显示结果
  7. plt.figure(figsize=(20, 8))
  8. img_ori = cv2.imread("./doc/imgs/00018069.jpg")
  9. img_out = cv2.imread("./inference_results/det_res_00018069.jpg")
  10. plt.subplot(1, 2, 1)
  11. plt.imshow(img_ori[:,:,::-1])
  12. plt.subplot(1, 2, 2)
  13. plt.imshow(img_out[:,:,::-1])
  14. plt.show()

首先需要定义参数设置如下所示。更多参数的介绍可以参考：PaddleOCR推理过程参数介绍。

  1. # 参考代码
  2. # https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.4/tools/infer/utility.py
  3. import argparse
  4. import os
  5. import sys
  6. import cv2
  7. import numpy as np
  8. import paddle
  9. from PIL import Image, ImageDraw, ImageFont
  10. import math
  11. from paddle import inference
  12. import time
  13. from ppocr.utils.logging import get_logger
  14.  
  15.  
  16. def str2bool(v):
  17.     return v.lower() in ("true", "t", "1")
  18.  
  19.  
  20. def init_args():
  21.     parser = argparse.ArgumentParser()
  22.     # params for prediction engine
  23.     parser.add_argument("--use_gpu", type=str2bool, default=True)
  24.     parser.add_argument("--ir_optim", type=str2bool, default=True)
  25.     parser.add_argument("--use_tensorrt", type=str2bool, default=False)
  26.     parser.add_argument("--min_subgraph_size", type=int, default=15)
  27.     parser.add_argument("--precision", type=str, default="fp32")
  28.     parser.add_argument("--gpu_mem", type=int, default=500)
  29.  
  30.     # params for text detector
  31.     parser.add_argument("--image_dir", type=str)
  32.     parser.add_argument("--det_algorithm", type=str, default='DB')
  33.     parser.add_argument("--det_model_dir", type=str)
  34.     parser.add_argument("--det_limit_side_len", type=float, default=960)
  35.     parser.add_argument("--det_limit_type", type=str, default='max')
  36.  
  37.     # DB parmas
  38.     parser.add_argument("--det_db_thresh", type=float, default=0.3)
  39.     parser.add_argument("--det_db_box_thresh", type=float, default=0.6)
  40.     parser.add_argument("--det_db_unclip_ratio", type=float, default=1.5)
  41.     parser.add_argument("--max_batch_size", type=int, default=10)
  42.     parser.add_argument("--use_dilation", type=str2bool, default=False)
  43.     parser.add_argument("--det_db_score_mode", type=str, default="fast")
  44.     # EAST parmas
  45.     parser.add_argument("--det_east_score_thresh", type=float, default=0.8)
  46.     parser.add_argument("--det_east_cover_thresh", type=float, default=0.1)
  47.     parser.add_argument("--det_east_nms_thresh", type=float, default=0.2)
  48.  
  49.     # SAST parmas
  50.     parser.add_argument("--det_sast_score_thresh", type=float, default=0.5)
  51.     parser.add_argument("--det_sast_nms_thresh", type=float, default=0.2)
  52.     parser.add_argument("--det_sast_polygon", type=str2bool, default=False)
  53.  
  54.     # PSE parmas
  55.     parser.add_argument("--det_pse_thresh", type=float, default=0)
  56.     parser.add_argument("--det_pse_box_thresh", type=float, default=0.85)
  57.     parser.add_argument("--det_pse_min_area", type=float, default=16)
  58.     parser.add_argument("--det_pse_box_type", type=str, default='box')
  59.     parser.add_argument("--det_pse_scale", type=int, default=1)
  60.  
  61.     # params for text recognizer
  62.     parser.add_argument("--rec_algorithm", type=str, default='CRNN')
  63.     parser.add_argument("--rec_model_dir", type=str)
  64.     parser.add_argument("--rec_image_shape", type=str, default="3, 32, 320")
  65.     parser.add_argument("--rec_batch_num", type=int, default=6)
  66.     parser.add_argument("--max_text_length", type=int, default=25)
  67.     parser.add_argument(
  68.         "--rec_char_dict_path",
  69.         type=str,
  70.         default="./ppocr/utils/ppocr_keys_v1.txt")
  71.     parser.add_argument("--use_space_char", type=str2bool, default=True)
  72.     parser.add_argument(
  73.         "--vis_font_path", type=str, default="./doc/fonts/simfang.ttf")
  74.     parser.add_argument("--drop_score", type=float, default=0.5)
  75.  
  76.     # params for e2e
  77.     parser.add_argument("--e2e_algorithm", type=str, default='PGNet')
  78.     parser.add_argument("--e2e_model_dir", type=str)
  79.     parser.add_argument("--e2e_limit_side_len", type=float, default=768)
  80.     parser.add_argument("--e2e_limit_type", type=str, default='max')
  81.  
  82.     # PGNet parmas
  83.     parser.add_argument("--e2e_pgnet_score_thresh", type=float, default=0.5)
  84.     parser.add_argument(
  85.         "--e2e_char_dict_path", type=str, default="./ppocr/utils/ic15_dict.txt")
  86.     parser.add_argument("--e2e_pgnet_valid_set", type=str, default='totaltext')
  87.     parser.add_argument("--e2e_pgnet_mode", type=str, default='fast')
  88.  
  89.     # params for text classifier
  90.     parser.add_argument("--use_angle_cls", type=str2bool, default=False)
  91.     parser.add_argument("--cls_model_dir", type=str)
  92.     parser.add_argument("--cls_image_shape", type=str, default="3, 48, 192")
  93.     parser.add_argument("--label_list", type=list, default=['0', '180'])
  94.     parser.add_argument("--cls_batch_num", type=int, default=6)
  95.     parser.add_argument("--cls_thresh", type=float, default=0.9)
  96.  
  97.     parser.add_argument("--enable_mkldnn", type=str2bool, default=False)
  98.     parser.add_argument("--cpu_threads", type=int, default=10)
  99.     parser.add_argument("--use_pdserving", type=str2bool, default=False)
  100.     parser.add_argument("--warmup", type=str2bool, default=False)
  101.  
  102.     #
  103.     parser.add_argument(
  104.         "--draw_img_save_dir", type=str, default="./inference_results")
  105.     parser.add_argument("--save_crop_res", type=str2bool, default=False)
  106.     parser.add_argument("--crop_res_save_dir", type=str, default="./output")
  107.  
  108.     # multi-process
  109.     parser.add_argument("--use_mp", type=str2bool, default=False)
  110.     parser.add_argument("--total_process_num", type=int, default=1)
  111.     parser.add_argument("--process_id", type=int, default=0)
  112.  
  113.     parser.add_argument("--benchmark", type=str2bool, default=False)
  114.     parser.add_argument("--save_log_path", type=str, default="./log_output/")
  115.  
  116.     parser.add_argument("--show_log", type=str2bool, default=True)
  117.     parser.add_argument("--use_onnx", type=str2bool, default=False)
  118.     # 这里需要注意，添加这个是因为直接在notebook中解析的话，sys.argv会在后面添加下面的内容，导致解析失败
  119.     # '-f', '/home/aistudio/.local/share/jupyter/runtime/kernel-e1221262-c656-4129-896f-1b197b6b782c.json'
  120.     parser.add_argument("-f", type=str, default=None)
  121.     return parser
  122.  
  123.  
  124. def parse_args():
  125.     parser = init_args()
  126.     return parser.parse_args()

下面和大家具体看下文字检测的具体代码。

  1. # 参考代码
  2. # https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.4/tools/infer/predict_det.py
  3. import os
  4. import sys
  5. import cv2
  6. import numpy as np
  7. import time
  8.  
  9. import tools.infer.utility as utility
  10. from ppocr.utils.logging import get_logger
  11. from ppocr.utils.utility import get_image_file_list, check_and_read_gif
  12. from ppocr.data import create_operators, transform
  13. from ppocr.postprocess import build_post_process
  14. import json
  15. logger = get_logger()
  16.  
  17. # 文字检测类
  18. class TextDetector(object):
  19.     def __init__(self, args):
  20.         self.args = args
  21.         self.det_algorithm = args.det_algorithm
  22.         pre_process_list = [{
  23.             'DetResizeForTest': {
  24.                 'limit_side_len': args.det_limit_side_len,
  25.                 'limit_type': args.det_limit_type,
  26.             }
  27.         }, {
  28.             'NormalizeImage': {
  29.                 'std': [0.229, 0.224, 0.225],
  30.                 'mean': [0.485, 0.456, 0.406],
  31.                 'scale': '1./255.',
  32.                 'order': 'hwc'
  33.             }
  34.         }, {
  35.             'ToCHWImage': None
  36.         }, {
  37.             'KeepKeys': {
  38.                 'keep_keys': ['image', 'shape']
  39.             }
  40.         }]
  41.         postprocess_params = {}
  42.         if self.det_algorithm == "DB":
  43.             postprocess_params['name'] = 'DBPostProcess'
  44.             postprocess_params["thresh"] = args.det_db_thresh
  45.             postprocess_params["box_thresh"] = args.det_db_box_thresh
  46.             postprocess_params["max_candidates"] = 1000
  47.             postprocess_params["unclip_ratio"] = args.det_db_unclip_ratio
  48.             postprocess_params["use_dilation"] = args.use_dilation
  49.             postprocess_params["score_mode"] = args.det_db_score_mode
  50.         else:
  51.             logger.info("unknown det_algorithm:{}".format(self.det_algorithm))
  52.             sys.exit(0)
  53.         # 初始化预测引擎
  54.         self.predictor, self.input_tensor, self.output_tensors, self.config = utility.create_predictor(
  55.             args, 'det', logger)
  56.         # 构建预处理算子
  57.         self.preprocess_op = create_operators(pre_process_list)
  58.         # 构建后处理算子
  59.         self.postprocess_op = build_post_process(postprocess_params)
  60.         
  61.  
  62.     def order_points_clockwise(self, pts):
  63.         """
  64.         参考: https://github.com/jrosebr1/imutils/blob/master/imutils/perspective.py
  65.         对检测出来的点进行按照顺时针排序
  66.         """
  67.         xSorted = pts[np.argsort(pts[:, 0]), :]
  68.  
  69.         leftMost = xSorted[:2, :]
  70.         rightMost = xSorted[2:, :]
  71.  
  72.         leftMost = leftMost[np.argsort(leftMost[:, 1]), :]
  73.         (tl, bl) = leftMost
  74.  
  75.         rightMost = rightMost[np.argsort(rightMost[:, 1]), :]
  76.         (tr, br) = rightMost
  77.  
  78.         rect = np.array([tl, tr, br, bl], dtype="float32")
  79.         return rect
  80.  
  81.     def clip_det_res(self, points, img_height, img_width):
  82.         # 对检测结果根据宽高进行限幅，防止超出图像边界
  83.         for pno in range(points.shape[0]):
  84.             points[pno, 0] = int(min(max(points[pno, 0], 0), img_width - 1))
  85.             points[pno, 1] = int(min(max(points[pno, 1], 0), img_height - 1))
  86.         return points
  87.  
  88.     def filter_tag_det_res(self, dt_boxes, image_shape):
  89.         # 去除小于特定尺寸的检测结果
  90.         img_height, img_width = image_shape[0:2]
  91.         dt_boxes_new = []
  92.         for box in dt_boxes:
  93.             box = self.order_points_clockwise(box)
  94.             box = self.clip_det_res(box, img_height, img_width)
  95.             rect_width = int(np.linalg.norm(box[0] - box[1]))
  96.             rect_height = int(np.linalg.norm(box[0] - box[3]))
  97.             if rect_width <= 3 or rect_height <= 3:
  98.                 continue
  99.             dt_boxes_new.append(box)
  100.         dt_boxes = np.array(dt_boxes_new)
  101.         return dt_boxes
  102.  
  103.     def filter_tag_det_res_only_clip(self, dt_boxes, image_shape):
  104.         # 仅对检测结果的边界进行限幅
  105.         img_height, img_width = image_shape[0:2]
  106.         dt_boxes_new = []
  107.         for box in dt_boxes:
  108.             box = self.clip_det_res(box, img_height, img_width)
  109.             dt_boxes_new.append(box)
  110.         dt_boxes = np.array(dt_boxes_new)
  111.         return dt_boxes
  112.  
  113.     def __call__(self, img):
  114.         ori_im = img.copy()
  115.         data = {'image': img}
  116.  
  117.         st = time.time()
  118.         
  119.         # 数据预处理
  120.         data = transform(data, self.preprocess_op)
  121.         img, shape_list = data
  122.         if img is None:
  123.             return None, 0
  124.         # 扩展bs维度：CHW -> NCHW
  125.         img = np.expand_dims(img, axis=0)
  126.         shape_list = np.expand_dims(shape_list, axis=0)
  127.         img = img.copy()
  128.         # 将数据拷贝到预测引擎中
  129.         self.input_tensor.copy_from_cpu(img)
  130.         # 自动推理
  131.         self.predictor.run()
  132.         outputs = []
  133.         # 将返回结果从预测引擎中拷贝回CPU
  134.         for output_tensor in self.output_tensors:
  135.             output = output_tensor.copy_to_cpu()
  136.             outputs.append(output)
  137.  
  138.         preds = {}
  139.         if self.det_algorithm in ['DB', 'PSE']:
  140.             preds['maps'] = outputs[0]
  141.         else:
  142.             raise NotImplementedError
  143.  
  144.         # 后处理
  145.         post_result = self.postprocess_op(preds, shape_list)
  146.         dt_boxes = post_result[0]['points']
  147.         dt_boxes = self.filter_tag_det_res(dt_boxes, ori_im.shape)
  148.         
  149.         et = time.time()
  150.         return dt_boxes, et - st
  151.  
  152. # 设置参数
  153. args = parse_args()
  154. args.det_model_dir = "./inference/ch_PP-OCRv2_det_infer"
  155. args.image_dir = "./doc/imgs/00018069.jpg"
  156.  
  157. # 获取图片列表
  158. image_file_list = get_image_file_list(args.image_dir)
  159. # 创建文本检测器对象
  160. text_detector = TextDetector(args)
  161.  
  162. count = 0
  163. total_time = 0
  164. draw_img_save = "./inference_results"
  165.  
  166. if not os.path.exists(draw_img_save):
  167.     os.makedirs(draw_img_save)
  168. save_results = []
  169. for image_file in image_file_list:
  170.     img = cv2.imread(image_file)
  171.     if img is None:
  172.         logger.info("error in loading image:{}".format(image_file))
  173.         continue
  174.     st = time.time()
  175.     dt_boxes, _ = text_detector(img)
  176.     elapse = time.time() - st
  177.     if count > 0:
  178.         total_time += elapse
  179.     count += 1
  180.     save_pred = os.path.basename(image_file) + "\t" + str(
  181.         json.dumps(np.array(dt_boxes).astype(np.int32).tolist())) + "\n"
  182.     save_results.append(save_pred)
  183.     logger.info(save_pred)
  184.     logger.info("The predict time of {}: {}".format(image_file, elapse))
  185.     src_im = utility.draw_text_det_res(dt_boxes, image_file)
  186.     img_name_pure = os.path.split(image_file)[-1]
  187.     img_path = os.path.join(draw_img_save,
  188.                             "det_res_{}".format(img_name_pure))
  189.     cv2.imwrite(img_path, src_im)
  190.     logger.info("The visualized image saved in {}".format(img_path))
  191.     
  192.     break
  193.  
  194. with open(os.path.join(draw_img_save, "det_results.txt"), 'w') as f:
  195.     f.writelines(save_results)
  196.     f.close()
  197.  
  198. plt.figure(figsize=(10, 10))
  199. plt.imshow(src_im[:, :, ::-1])
  200. plt.show()

上面就完成了完整的文本检测流程。

2.3  2.3 PP-OCRv2方向分类器模型推理

同样地，对于方向分类器模型，我们也可以使用下面的命令快速体验其功能。

  1. # 下载模型
  2. !cd inference && wget https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_cls_infer.tar -O ch_ppocr_mobile_v2.0_cls_infer.tar && tar -xf ch_ppocr_mobile_v2.0_cls_infer.tar
  3. # 预测
  4. !python tools/infer/predict_cls.py \
  5.     --image_dir="./doc/imgs_words/ch/word_1.jpg" \
  6.     --cls_model_dir="./inference/ch_ppocr_mobile_v2.0_cls_infer" \
  7.     --use_gpu=False
  8. # 画图
  9. img = cv2.imread("./doc/imgs_words/ch/word_1.jpg")
  10. plt.imshow(img[:,:,::-1])
  11. plt.show()

图片的方向是正向水平文本，预测结果正确。

具体地，方向分类器的具体实现代码如下所示。

  1. # 参考代码
  2. # https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.4/tools/infer/predict_cls.py
  3. import copy
  4.  
  5. # 方向分类器实现的类
  6. class TextClassifier(object):
  7.     def __init__(self, args):
  8.         self.cls_image_shape = [int(v) for v in args.cls_image_shape.split(",")]
  9.         self.cls_batch_num = args.cls_batch_num
  10.         self.cls_thresh = args.cls_thresh
  11.         postprocess_params = {
  12.             'name': 'ClsPostProcess',
  13.             "label_list": args.label_list,
  14.         }
  15.         # 后处理算子
  16.         self.postprocess_op = build_post_process(postprocess_params)
  17.         # 初始化预测引擎
  18.         self.predictor, self.input_tensor, self.output_tensors, _ = \
  19.             utility.create_predictor(args, 'cls', logger)
  20.  
  21.     # 对图像进行resize并且normalize
  22.     def resize_norm_img(self, img):
  23.         imgC, imgH, imgW = self.cls_image_shape
  24.         h = img.shape[0]
  25.         w = img.shape[1]
  26.         ratio = w / float(h)
  27.         if math.ceil(imgH * ratio) > imgW:
  28.             resized_w = imgW
  29.         else:
  30.             resized_w = int(math.ceil(imgH * ratio))
  31.         resized_image = cv2.resize(img, (resized_w, imgH))
  32.         resized_image = resized_image.astype('float32')
  33.         if self.cls_image_shape[0] == 1:
  34.             resized_image = resized_image / 255
  35.             resized_image = resized_image[np.newaxis, :]
  36.         else:
  37.             resized_image = resized_image.transpose((2, 0, 1)) / 255
  38.         resized_image -= 0.5
  39.         resized_image /= 0.5
  40.         padding_im = np.zeros((imgC, imgH, imgW), dtype=np.float32)
  41.         padding_im[:, :, 0:resized_w] = resized_image
  42.         return padding_im
  43.  
  44.     def __call__(self, img_list):
  45.         img_list = copy.deepcopy(img_list)
  46.         img_num = len(img_list)
  47.         # 记录宽高比
  48.         width_list = []
  49.         for img in img_list:
  50.             width_list.append(img.shape[1] / float(img.shape[0]))
  51.         # 排序，加速后续的预处理过程
  52.         indices = np.argsort(np.array(width_list))
  53.  
  54.         cls_res = [['', 0.0]] * img_num
  55.         batch_num = self.cls_batch_num
  56.         elapse = 0
  57.         for beg_img_no in range(0, img_num, batch_num):
  58.             end_img_no = min(img_num, beg_img_no + batch_num)
  59.             norm_img_batch = []
  60.             max_wh_ratio = 0
  61.             starttime = time.time()
  62.             # 预处理数据，组batch
  63.             for ino in range(beg_img_no, end_img_no):
  64.                 h, w = img_list[indices[ino]].shape[0:2]
  65.                 wh_ratio = w * 1.0 / h
  66.                 max_wh_ratio = max(max_wh_ratio, wh_ratio)
  67.             for ino in range(beg_img_no, end_img_no):
  68.                 norm_img = self.resize_norm_img(img_list[indices[ino]])
  69.                 norm_img = norm_img[np.newaxis, :]
  70.                 norm_img_batch.append(norm_img)
  71.             norm_img_batch = np.concatenate(norm_img_batch)
  72.             norm_img_batch = norm_img_batch.copy()
  73.             # 将数据拷贝到预测引擎
  74.             self.input_tensor.copy_from_cpu(norm_img_batch)
  75.             # 自动推理过程
  76.             self.predictor.run()
  77.             # 将数据拷贝回CPU
  78.             prob_out = self.output_tensors[0].copy_to_cpu()
  79.             # 后处理
  80.             cls_result = self.postprocess_op(prob_out)
  81.             elapse += time.time() - starttime
  82.             for rno in range(len(cls_result)):
  83.                 label, score = cls_result[rno]
  84.                 cls_res[indices[beg_img_no + rno]] = [label, score]
  85.                 if '180' in label and score > self.cls_thresh:
  86.                     img_list[indices[beg_img_no + rno]] = cv2.rotate(
  87.                         img_list[indices[beg_img_no + rno]], 1)
  88.         return img_list, cls_res, elapse
  89.  
  90. args = parse_args()
  91. args.cls_model_dir = "./inference/ch_ppocr_mobile_v2.0_cls_infer"
  92. args.image_dir = "./doc/imgs_words/ch/word_4.jpg"
  93.  
  94. image_file_list = get_image_file_list(args.image_dir)
  95. text_classifier = TextClassifier(args)
  96. valid_image_file_list = []
  97. img_list = []
  98. for image_file in image_file_list:
  99.     img = cv2.imread(image_file)
  100.     # 预测之前对图像旋转180度
  101.     # img = cv2.rotate(img, cv2.ROTATE_180)
  102.     if img is None:
  103.         logger.info("error in loading image:{}".format(image_file))
  104.         continue
  105.     valid_image_file_list.append(image_file)
  106.     img_list.append(img)
  107. img_list, cls_res, predict_time = text_classifier(img_list)
  108. for ino in range(len(img_list)):
  109.     logger.info("Predicts of {}:{}".format(valid_image_file_list[ino],
  110.                                             cls_res[ino]))
  111.  
  112. plt.imshow(img[:,:,::-1])
  113. plt.show()

这里我们也可以将图像旋转180度之后，看下方向分类器的分类效果。

上面就完成了完整的方向分类器的推理过程。

2.4 PP-OCRv2文字识别模型推理

对于文字识别模型，我们也可以使用下面的命令快速体验其功能。

  1. # 下载模型
  2. !cd inference && wget https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_rec_infer.tar -O ch_PP-OCRv2_rec_infer.tar &&  tar -xf ch_PP-OCRv2_rec_infer.tar
  3. # 预测
  4. !python tools/infer/predict_rec.py \
  5.     --image_dir="./doc/imgs_words/ch/word_4.jpg" \
  6.     --rec_model_dir="./inference/ch_PP-OCRv2_rec_infer" \
  7.     --use_gpu=False
  8.  
  9. # 读取图像并显示
  10. img = cv2.imread("./doc/imgs_words/ch/word_4.jpg")
  11. plt.imshow(img[:,:,::-1])
  12. plt.show()

文字识别的具体代码如下所示。

  1. # 参考代码
  2. # https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.4/tools/infer/predict_rec.py
  3. class TextRecognizer(object):
  4.     def __init__(self, args):
  5.         self.rec_image_shape = [int(v) for v in args.rec_image_shape.split(",")]
  6.         self.rec_batch_num = args.rec_batch_num
  7.         self.rec_algorithm = args.rec_algorithm
  8.         postprocess_params = {
  9.             'name': 'CTCLabelDecode',
  10.             "character_dict_path": args.rec_char_dict_path,
  11.             "use_space_char": args.use_space_char
  12.         }
  13.         # 初始化预测引擎
  14.         self.predictor, self.input_tensor, self.output_tensors, self.config = \
  15.             utility.create_predictor(args, 'rec', logger)
  16.         # 初始化后处理过程
  17.         self.postprocess_op = build_post_process(postprocess_params)
  18.  
  19.     # 预处理核心逻辑
  20.     def resize_norm_img(self, img, max_wh_ratio):
  21.         imgC, imgH, imgW = self.rec_image_shape
  22.         assert imgC == img.shape[2]
  23.         imgW = int((32 * max_wh_ratio))
  24.         h, w = img.shape[:2]
  25.         ratio = w / float(h)
  26.         if math.ceil(imgH * ratio) > imgW:
  27.             resized_w = imgW
  28.         else:
  29.             resized_w = int(math.ceil(imgH * ratio))
  30.         resized_image = cv2.resize(img, (resized_w, imgH))
  31.         resized_image = resized_image.astype('float32')
  32.         # [0, 255] -> [0, 1]
  33.         resized_image = resized_image.transpose((2, 0, 1)) / 255
  34.         # [0, 1] -> [-0.5, 0.5]
  35.         resized_image -= 0.5
  36.         # [-0.5, 0.5] -> [-1, 1]
  37.         resized_image /= 0.5
  38.         padding_im = np.zeros((imgC, imgH, imgW), dtype=np.float32)
  39.         padding_im[:, :, 0:resized_w] = resized_image
  40.         return padding_im
  41.  
  42.     # 对图像列表进行处理
  43.     def __call__(self, img_list):
  44.         img_num = len(img_list)
  45.         # 记录宽高比
  46.         width_list = []
  47.         for img in img_list:
  48.             width_list.append(img.shape[1] / float(img.shape[0]))
  49.         # 排序，加速处理过程
  50.         indices = np.argsort(np.array(width_list))
  51.         rec_res = [['', 0.0]] * img_num
  52.         batch_num = self.rec_batch_num
  53.         st = time.time()
  54.         for beg_img_no in range(0, img_num, batch_num):
  55.             end_img_no = min(img_num, beg_img_no + batch_num)
  56.             norm_img_batch = []
  57.             max_wh_ratio = 0
  58.             for ino in range(beg_img_no, end_img_no):
  59.                 h, w = img_list[indices[ino]].shape[0:2]
  60.                 wh_ratio = w * 1.0 / h
  61.                 max_wh_ratio = max(max_wh_ratio, wh_ratio)
  62.             # 调用预处理方法并组batch
  63.             for ino in range(beg_img_no, end_img_no):
  64.                 norm_img = self.resize_norm_img(img_list[indices[ino]],
  65.                                                 max_wh_ratio)
  66.                 norm_img = norm_img[np.newaxis, :]
  67.                 norm_img_batch.append(norm_img)
  68.             norm_img_batch = np.concatenate(norm_img_batch)
  69.             norm_img_batch = norm_img_batch.copy()
  70.             
  71.             # 将数据拷贝到预测引擎中
  72.             self.input_tensor.copy_from_cpu(norm_img_batch)
  73.             # 自动化推理过程
  74.             self.predictor.run()
  75.             outputs = []
  76.             # 将数据拷贝到CPU
  77.             for output_tensor in self.output_tensors:
  78.                 output = output_tensor.copy_to_cpu()
  79.                 outputs.append(output)
  80.             if len(outputs) != 1:
  81.                 preds = outputs
  82.             else:
  83.                 preds = outputs[0]
  84.             # 后处理
  85.             rec_result = self.postprocess_op(preds)
  86.             for rno in range(len(rec_result)):
  87.                 rec_res[indices[beg_img_no + rno]] = rec_result[rno]
  88.         return rec_res, time.time() - st
  89.  
  90.  
  91. # 定义参数
  92. args = parse_args()
  93. args.rec_model_dir = "./inference/ch_PP-OCRv2_rec_infer"
  94. args.image_dir = "./doc/imgs_words/ch/word_4.jpg"
  95. img_list = []
  96.  
  97. image_file_list = get_image_file_list(args.image_dir)
  98. text_recognizer = TextRecognizer(args)
  99. valid_image_file_list = []
  100. for image_file in image_file_list:
  101.     img = cv2.imread(image_file)
  102.     if img is None:
  103.         logger.info("error in loading image:{}".format(image_file))
  104.         continue
  105.     valid_image_file_list.append(image_file)
  106.     img_list.append(img)
  107. rec_res, _ = text_recognizer(img_list)
  108. for ino in range(len(img_list)):
  109.     logger.info("Predicts of {}:{}".format(valid_image_file_list[ino],
  110.                                             rec_res[ino]))

2.5 PP-OCRv2 系统串联预测推理

前面的内容给大家介绍了PP-OCRv2系统中，检测、方向分类器、识别模型的单独推理过程。为了方便大家端到端地使用，我们将这三个模块串联起来，组成了PP-OCRv2系统，并提供了相应的预测脚本。

在执行PP-OCRv2的系统推理时，需要通过参数image_dir指定单张图像或者图像集合的路径、参数det_model_dir, cls_model_dir 和 rec_model_dir 分别指定检测、方向分类和识别的 inference 模型路径。参数 use_angle_cls 用于控制是否启用方向分类模型。use_mp 表示是否使用多进程。total_process_num 表示在使用多进程时的进程数。

以图像文件./doc/imgs/00018069.jpg为例，预测的原始图像如下。

如果串联预测时使用方向分类器，则可以使用下面的命令进行预测。

  1. # 使用方向分类器，运行PP-OCRv2系统
  2. !python tools/infer/predict_system.py \
  3.     --image_dir="./doc/imgs/00018069.jpg" \
  4.     --det_model_dir="./inference/ch_PP-OCRv2_det_infer/" \
  5.     --cls_model_dir="./inference/ch_ppocr_mobile_v2.0_cls_infer/" \
  6.     --rec_model_dir="./inference/ch_PP-OCRv2_rec_infer/" \
  7.     --use_angle_cls=True
  8.  
  9. # 可视化
  10. img = cv2.imread("./inference_results/00018069.jpg")
  11. plt.figure(figsize=(20, 8))
  12. plt.imshow(img[..., ::-1])
  13. plt.show()

可视化识别结果默认保存到 ./inference_results 文件夹里面。

在图象中可视化出了检测框和识别结果，在上面的notebook中也打印出了具体的识别文件以及文件读取路径信息。

如果希望保存裁剪后的识别结果，可以将save_crop_res参数设置为True，最终结果保存在output目录下，其中部分裁剪后图像如下所示。保存的结果可以用于后续的识别模型标注与训练。

  1. # 裁剪文字检测的结果图像并保存
  2. !python tools/infer/predict_system.py \
  3.     --image_dir="./doc/imgs/00018069.jpg" \
  4.     --det_model_dir="./inference/ch_PP-OCRv2_det_infer/" \
  5.     --cls_model_dir="./inference/ch_ppocr_mobile_v2.0_cls_infer/" \
  6.     --rec_model_dir="./inference/ch_PP-OCRv2_rec_infer/" \
  7.     --use_angle_cls=True \
  8.     --save_crop_res=True
  9.  
  10. !ls output
  11.  
  12. plt.figure(figsize=(8, 8))
  13. plt.imshow(cv2.imread("./doc/imgs/00018069.jpg")[:, :, ::-1])
  14. plt.show()
  15. plt.figure(figsize=(14, 4))
  16. plt.subplot(1,3,1)
  17. plt.imshow(cv2.imread("output/mg_crop_0.jpg")[:, :, ::-1])
  18. plt.subplot(1,3,2)
  19. plt.imshow(cv2.imread("output/mg_crop_1.jpg")[:, :, ::-1])
  20. plt.subplot(1,3,3)
  21. plt.imshow(cv2.imread("output/mg_crop_2.jpg")[:, :, ::-1])
  22. plt.show()

串联预测通过TextSystem类进行实现，其具体实现过程与函数定义如下。

  1. # 参考代码：https://github.com/PaddlePaddle/PaddleOCR/blob/release%2F2.4/tools/infer/predict_system.py
  2. from tools.infer.utility import draw_ocr_box_txt, get_rotate_crop_image
  3. from ppocr.utils.utility import get_image_file_list
  4.  
  5. class TextSystem(object):
  6.     # 初始化函数
  7.     def __init__(self, args):
  8.         self.args = args
  9.   		# 如果不希望显示log，可以将show_log设置为False
  10.         if not args.show_log:
  11.             logger.setLevel(logging.INFO)
  12.         # 定义文本检测模型预测引擎
  13.         self.text_detector = TextDetector(args)
  14.         # 定义文本识别模型预测引擎
  15.         self.text_recognizer = TextRecognizer(args)
  16.         # 是否使用方向分类器
  17.         self.use_angle_cls = args.use_angle_cls
  18.         # 得分阈值，根据该阈值判断检测与识别结果是否需要进行可视化或者返回
  19.         self.drop_score = args.drop_score
  20.         # 定义方向分类器预测引擎
  21.         if self.use_angle_cls:
  22.             self.text_classifier = TextClassifier(args)
  23. 	
  24.   	# 保存文本检测结果图像
  25.     def draw_crop_rec_res(self, output_dir, img_crop_list, rec_res):
  26.         os.makedirs(output_dir, exist_ok=True)
  27.         bbox_num = len(img_crop_list)
  28.         for bno in range(bbox_num):
  29.             cv2.imwrite(
  30.                 os.path.join(output_dir,
  31.                              f"mg_crop_{bno+self.crop_image_res_index}.jpg"),
  32.                 img_crop_list[bno])
  33.             logger.debug(f"{bno}, {rec_res[bno]}")
  34.         self.crop_image_res_index += bbox_num
  35. 	
  36.     # 核心预测函数
  37.     def __call__(self, img, cls=True):
  38.         ori_im = img.copy()
  39.         # 获取检测文本检测结果
  40.         dt_boxes, elapse = self.text_detector(img)
  41.         logger.debug("dt_boxes num : {}, elapse : {}".format(
  42.             len(dt_boxes), elapse))
  43.         if dt_boxes is None:
  44.             return None, None
  45.         img_crop_list = []
  46.         # 对检测框进行排序，顺序为：优先从上到下，其次从左到右
  47.         dt_boxes = sorted_boxes(dt_boxes)
  48.         # 对检测结果进行透视变换与校正
  49.         for bno in range(len(dt_boxes)):
  50.             tmp_box = copy.deepcopy(dt_boxes[bno])
  51.             img_crop = get_rotate_crop_image(ori_im, tmp_box)
  52.             img_crop_list.append(img_crop)
  53.         # 使用方向分类器对检测结果进行转正
  54.         if self.use_angle_cls and cls:
  55.             img_crop_list, angle_list, elapse = self.text_classifier(
  56.                 img_crop_list)
  57.             logger.debug("cls num  : {}, elapse : {}".format(
  58.                 len(img_crop_list), elapse))
  59.         # 获取文本识别结果
  60.         rec_res, elapse = self.text_recognizer(img_crop_list)
  61.         logger.debug("rec_res num  : {}, elapse : {}".format(
  62.             len(rec_res), elapse))
  63.         # 保存经过校正之后的文本检测图像
  64.         if self.args.save_crop_res:
  65.             self.draw_crop_rec_res(self.args.crop_res_save_dir, img_crop_list,
  66.                                    rec_res)
  67.         filter_boxes, filter_rec_res = [], []
  68.         # 根据识别得分的阈值对结果进行过滤，如果得分小于阈值，就过滤掉
  69.         for box, rec_result in zip(dt_boxes, rec_res):
  70.             text, score = rec_result
  71.             if score >= self.drop_score:
  72.                 filter_boxes.append(box)
  73.                 filter_rec_res.append(rec_result)
  74.         return filter_boxes, filter_rec_res
  75.  
  76. def sorted_boxes(dt_boxes):
  77.     # 对检测框进行排序：优先从上到下，其次从左到右
  78.     num_boxes = dt_boxes.shape[0]
  79.     sorted_boxes = sorted(dt_boxes, key=lambda x: (x[0][1], x[0][0]))
  80.     _boxes = list(sorted_boxes)
  81.  
  82.     for i in range(num_boxes - 1):
  83.         if abs(_boxes[i + 1][0][1] - _boxes[i][0][1]) < 10 and \
  84.                 (_boxes[i + 1][0][0] < _boxes[i][0][0]):
  85.             tmp = _boxes[i]
  86.             _boxes[i] = _boxes[i + 1]
  87.             _boxes[i + 1] = tmp
  88.     return _boxes
  89.  
  90. args = parse_args()
  91. args.cls_model_dir = "./inference/ch_ppocr_mobile_v2.0_cls_infer"
  92. args.det_model_dir="./inference/ch_PP-OCRv2_det_infer/"
  93. args.rec_model_dir="./inference/ch_PP-OCRv2_rec_infer/"
  94. args.image_dir = "./doc/imgs/00018069.jpg"
  95. args.use_angle_cls=True
  96. args.use_gpu=True
  97.  
  98. image_file_list = get_image_file_list(args.image_dir)
  99. image_file_list = image_file_list[args.process_id::args.total_process_num]
  100. text_sys = TextSystem(args)
  101. is_visualize = True
  102. font_path = args.vis_font_path
  103. drop_score = args.drop_score
  104.  
  105. total_time = 0
  106. cpu_mem, gpu_mem, gpu_util = 0, 0, 0
  107. _st = time.time()
  108. count = 0
  109. for idx, image_file in enumerate(image_file_list):
  110.     img = cv2.imread(image_file)
  111.     if img is None:
  112.         logger.debug("error in loading image:{}".format(image_file))
  113.         continue
  114.     starttime = time.time()
  115.     dt_boxes, rec_res = text_sys(img)
  116.     elapse = time.time() - starttime
  117.     total_time += elapse
  118.  
  119.     logger.debug(
  120.         str(idx) + "  Predict time of %s: %.3fs" % (image_file, elapse))
  121.     for text, score in rec_res:
  122.         logger.debug("{}, {:.3f}".format(text, score))
  123.  
  124.     if is_visualize:
  125.         image = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
  126.         boxes = dt_boxes
  127.         txts = [rec_res[i][0] for i in range(len(rec_res))]
  128.         scores = [rec_res[i][1] for i in range(len(rec_res))]
  129.  
  130.         draw_img = draw_ocr_box_txt(
  131.             image,
  132.             boxes,
  133.             txts,
  134.             scores,
  135.             drop_score=drop_score,
  136.             font_path=font_path)
  137.         draw_img_save_dir = args.draw_img_save_dir
  138.         os.makedirs(draw_img_save_dir, exist_ok=True)
  139.         cv2.imwrite(
  140.             os.path.join(draw_img_save_dir, os.path.basename(image_file)),
  141.             draw_img[:, :, ::-1])
  142.         logger.debug("The visualized image saved in {}".format(
  143.             os.path.join(draw_img_save_dir, os.path.basename(image_file))))
  144.  
  145. logger.info("The predict total time is {}".format(time.time() - _st))
  146.  
  147. plt.figure(figsize=(8, 8))
  148. plt.imshow(image)
  149. plt.show()
  150. plt.figure(figsize=(16, 8))
  151. plt.imshow(draw_img)
  152. plt.show()

2.6 PP-OCRv2 使用whl包预测推理

为了更加方便快速体验OCR文本检测与识别模型，PaddleOCR提供了基于Paddle Inference预测引擎的whl包，方便您一键安装，体验PaddleOCR。

2.6.1 安装whl包

使用pip安装paddleocr的whl包，命令如下。

  1. !pip install "paddleocr==2.3.0.2"
  2.  
  3. # 如果希望获取最新特性，可以基于源码编译安装
  4. #     python3 setup.py bdist_wheel
  5. #     pip3 install dist/paddleocr-x.x.x-py3-none-any.whl # x.x.x是paddleocr的版本号

2.6.2 使用whl包预测推理

paddleocr whl包会自动下载PP-OCRv2超轻量模型作为默认模型，也支持自定义模型路径、预测配置等参数，参数名称与基于Paddle Inference的python预测中参数相同。

  • 单独执行检测。

运行下面的代码，可快速体验文本检测模型的预测与效果。

  1. from paddleocr import PaddleOCR, draw_ocr
  2.  
  3. ocr = PaddleOCR(use_gpu=False)  # need to run only once to download and load model into memory
  4. img_path = '/home/aistudio/PaddleOCR/doc/imgs/11.jpg'
  5. result = ocr.ocr(img_path, rec=False)
  6. for line in result:
  7.     print(line)
  8.  
  9. # 显示结果
  10. from PIL import Image
  11.  
  12. image = Image.open(img_path).convert('RGB')
  13. im_show = draw_ocr(image, result, txts=None, scores=None, font_path='/home/aistudio/PaddleOCR/doc/fonts/simfang.ttf')
  14. plt.figure(figsize=(15, 8))
  15. plt.imshow(im_show)
  16. plt.show()
  • 单独执行识别

可以指定det=False，仅运行单独的识别模块。

  1. from paddleocr import PaddleOCR
  2.  
  3. ocr = PaddleOCR(use_gpu=False)  # need to run only once to download and load model into memory
  4. img_path = '/home/aistudio/PaddleOCR/doc/imgs_words/ch/word_1.jpg'
  5. result = ocr.ocr(img_path, det=False)
  6. for line in result:
  7.     print(line)
  • 单独执行方向分类器

可以指定det=False, rec=False, cls=True，仅运行方向分类器。

  1. from paddleocr import PaddleOCR
  2.  
  3. ocr = PaddleOCR(use_angle_cls=True, use_gpu=False)  # need to run only once to download and load model into memory
  4. img_path = '/home/aistudio/PaddleOCR/doc/imgs_words/ch/word_1.jpg'
  5. result = ocr.ocr(img_path, det=False, rec=False, cls=True)
  6. for line in result:
  7.     print(line)
  8.  
  9. img = cv2.imread(img_path)
  10. plt.imshow(img[...,::-1])
  11. plt.show()
  • 检测+方向分类器+识别全流程体验
  1. from paddleocr import PaddleOCR, draw_ocr
  2. import matplotlib.pyplot as plt
  3. %matplotlib inline
  4.  
  5. # PaddleOCR目前支持中英文、英文、法语、德语、韩语、日语，可以通过修改lang参数进行切换
  6. # 参数依次为`ch`, `en`, `french`, `german`, `korean`, `japan`。
  7. ocr = PaddleOCR(use_angle_cls=True, lang="ch", use_gpu=False)  # need to run only once to download and load model into memory
  8. img_path = '/home/aistudio/PaddleOCR/doc/imgs/11.jpg'
  9. result = ocr.ocr(img_path, cls=True)
  10. for line in result:
  11.     print(line)
  12.  
  13. # 显示结果
  14. from PIL import Image
  15.  
  16. image = Image.open(img_path).convert('RGB')
  17. boxes = [line[0] for line in result]
  18. txts = [line[1][0] for line in result]
  19. scores = [line[1][1] for line in result]
  20. im_show = draw_ocr(image, boxes, txts, scores, font_path='/home/aistudio/PaddleOCR/doc/fonts/simfang.ttf')
  21. plt.figure(figsize=(15, 8))
  22. plt.imshow(im_show)
  23. plt.show()

结果是一个list，每个item包含了文本框，文字和识别置信度。

3 基于Paddle Inference的C++推理

在推理部署过程中，相比于Python，C++的性能一般会更好一些，因此很多实际推理场景中会考虑使用C++作为开发语言进行推理。

上一小节给大家介绍的Paddle Inference也支持C++的推理过程，本节主要介绍C++的PP-OCRv2推理过程。

在基于Paddle Inference，对PP-OCRv2系统使用C++推理时，有以下几个步骤。

（1）准备模型

（2）编译opencv库

（3）获取Paddle Inference预测库

（4）编译PaddleOCR C++推理代码

（5）运行PP-OCRv2系统

由于AiStudio上版本限制，这里不做具体演示过程，仅给大家介绍具体的流程。建议您在本地体验PP-OCRv2的C++推理过程。

关于本小节更加详细的内容可以参考：PP-OCRv2 C++推理教程。

3.1 准备模型

使用下面的命令，准备PP-OCRv2的推理模型。

  1. cd deploy/cpp_infer
  2. wget https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_det_infer.tar -O ch_PP-OCRv2_det_infer.tar && tar -xf ch_PP-OCRv2_det_infer.tar
  3. wget https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_rec_infer.tar -O ch_PP-OCRv2_rec_infer.tar &&  tar -xf ch_PP-OCRv2_rec_infer.tar
  4. wget https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_cls_infer.tar -O ch_ppocr_mobile_v2.0_cls_infer.tar && tar -xf ch_ppocr_mobile_v2.0_cls_infer.tar

3.2 编译OpenCV库

  • 首先需要从opencv官网上下载在Linux环境下源码编译的包，以opencv3.4.7为例，下载命令如下。
  1. wget https://paddleocr.bj.bcebos.com/libs/opencv/opencv-3.4.7.tar.gz
  2. tar -xf opencv-3.4.7.tar.gz

最终可以在当前目录下看到opencv-3.4.7/的文件夹。

  • 编译OpenCV，设置OpenCV源码路径(root_path)以及安装路径(install_path)。进入opencv源码路径下，按照下面的方式进行编译。
  1. root_path="your_opencv_root_path"
  2. install_path=${root_path}/opencv3
  3. build_dir=${root_path}/build
  4.  
  5. rm -rf ${build_dir}
  6. mkdir ${build_dir}
  7. cd ${build_dir}
  8.  
  9. cmake .. \
  10.     -DCMAKE_INSTALL_PREFIX=${install_path} \
  11.     -DCMAKE_BUILD_TYPE=Release \
  12.     -DBUILD_SHARED_LIBS=OFF \
  13.     -DWITH_IPP=OFF \
  14.     -DBUILD_IPP_IW=OFF \
  15.     -DWITH_LAPACK=OFF \
  16.     -DWITH_EIGEN=OFF \
  17.     -DCMAKE_INSTALL_LIBDIR=lib64 \
  18.     -DWITH_ZLIB=ON \
  19.     -DBUILD_ZLIB=ON \
  20.     -DWITH_JPEG=ON \
  21.     -DBUILD_JPEG=ON \
  22.     -DWITH_PNG=ON \
  23.     -DBUILD_PNG=ON \
  24.     -DWITH_TIFF=ON \
  25.     -DBUILD_TIFF=ON
  26.  
  27. make -j
  28. make install

也可以直接修改tools/build_opencv.sh的内容，然后直接运行下面的命令进行编译。

sh tools/build_opencv.sh

其中root_path为下载的opencv源码路径，install_path为opencv的安装路径，make install完成之后，会在该文件夹下生成opencv头文件和库文件，用于后面的OCR代码编译。

最终在安装路径下的文件结构如下所示。

  1. opencv3/
  2. |-- bin
  3. |-- include
  4. |-- lib
  5. |-- lib64
  6. |-- share

3.3 获取Paddle Inference预测库

  • Paddle预测库官网 上提供了不同cuda版本的Linux预测库，可以在官网根据自己的环境选择合适的预测库版本。

  • 下载之后使用下面的方法解压。

  1. wget https://paddle-inference-lib.bj.bcebos.com/2.2.1/cxx_c/Linux/GPU/x86-64_gcc8.2_avx_mkl_cuda10.2_cudnn8.1.1_trt7.2.3.4/paddle_inference.tgz -O paddle_inference.tgz
  2. tar -xf paddle_inference.tgz

最终会在当前的文件夹中生成paddle_inference/的子文件夹。

3.4 编译PaddleOCR的推理代码

编译命令如下，其中Paddle C++预测库、opencv等其他依赖库的地址需要换成自己机器上的实际地址。

sh tools/build.sh
  • 具体的，需要修改tools/build.sh中环境路径，相关内容如下：
  1. OPENCV_DIR=your_opencv_dir
  2. LIB_DIR=your_paddle_inference_dir
  3. CUDA_LIB_DIR=your_cuda_lib_dir
  4. CUDNN_LIB_DIR=/your_cudnn_lib_dir

其中，OPENCV_DIR为opencv编译安装的地址；LIB_DIR为下载(paddle_inference文件夹)或者编译生成的Paddle预测库地址(build/paddle_inference_install_dir文件夹)；CUDA_LIB_DIR为cuda库文件地址，在docker中为/usr/local/cuda/lib64；CUDNN_LIB_DIR为cudnn库文件地址，在docker中为/usr/lib/x86_64-linux-gnu/。注意：以上路径都写绝对路径，不要写相对路径。

  • 编译完成之后，会在build文件夹下生成一个名为ppocr的可执行文件。

3.5 运行PP-OCRv2系统

运行方式：

./build/ppocr <mode> [--param1] [--param2] [...]

其中，mode为必选参数，表示选择的功能，取值范围['det', 'rec', 'system']，分别表示调用检测、识别、检测识别串联（包括方向分类器）。具体命令如下：

  • 只运行文字检测模型
  1. ./build/ppocr det \
  2.     --det_model_dir=./ch_PP-OCRv2_det_infer/ \
  3.     --image_dir=../../doc/imgs/12.jpg
  • 只运行文字识别模型
  1. ./build/ppocr rec \
  2.     --rec_model_dir=./ch_PP-OCRv2_rec_infer/ \
  3.     --image_dir=../../doc/imgs_words/ch/
  • 运行PP-OCRv2系统
  1. # 不使用方向分类器
  2. ./build/ppocr system \
  3.     --det_model_dir=./ch_PP-OCRv2_det_infer/ \
  4.     --rec_model_dir=./ch_PP-OCRv2_rec_infer/ \
  5.     --image_dir=../../doc/imgs/12.jpg
  6.  
  7. # 使用方向分类器
  8. ./build/ppocr system \
  9.     --det_model_dir=./ch_PP-OCRv2_det_infer/ \
  10.     --rec_model_dir=./ch_PP-OCRv2_rec_infer/ \
  11.     --use_angle_cls=true \
  12.     --cls_model_dir=./ch_ppocr_mobile_v2.0_cls_infer \
  13.     --image_dir=../../doc/imgs/12.jpg

4 Paddle Serving服务化部署实战

在第2和第3节内容中，我们详细介绍了基于Paddle Inference的PP-OCRv2系统推理，它属于离线推理，即在特定机器上部署的代码只能在这台机器上使用，无法通过其他机器进行访问。因此模型服务化部署的需求也就衍生出来。

服务化部署指的是，将模型以服务的形式进行部署，其他的设备可以通过发送请求的形式去访问服务，从而获取模型服务的推理结果。服务化部署示意图如下所示。

在模型部署成功后，不同用户都可以通过客户端，以发送网络请求的方式获得推理服务。

Paddle Serving是飞桨为方便开发者进行服务化部署而打造的工具，本节主要介绍基于Paddle Serving的PP-OCRv2系统服务化部署过程。

4.1 Paddle Serving简介

Paddle Serving作为飞桨（PaddlePaddle）开源的服务化部署框架，长期目标就是围绕着人工智能落地的最后一公里提供越来越专业、可靠、易用的服务。Paddle Serving目前提供了两套框架C++ Serving和Python Pipeline。Python Pipeline框架倾向于二次开发的便捷性，C++ Serving框架更倾向于追求极致性能。

基于Paddle Serving进行PP-OCRv2模型的服务化部署时，流程如下所示。

4.2 准备预测数据和部署环境

数据与模型推理所用数据一致。

运行Paddle Serving，需要安装Paddle Serving三个安装包：paddle-serving-server、paddle-serving-client 和 paddle-serving-app，命令如下。

  1. !wget https://paddle-serving.bj.bcebos.com/test-dev/whl/paddle_serving_server_gpu-0.7.0.post102-py3-none-any.whl
  2. !pip install paddle_serving_server_gpu-0.7.0.post102-py3-none-any.whl
  3.  
  4. !wget https://paddle-serving.bj.bcebos.com/test-dev/whl/paddle_serving_client-0.7.0-cp37-none-any.whl
  5. !pip install paddle_serving_client-0.7.0-cp37-none-any.whl
  6.  
  7. !wget https://paddle-serving.bj.bcebos.com/test-dev/whl/paddle_serving_app-0.7.0-py3-none-any.whl
  8. !pip install paddle_serving_app-0.7.0-py3-none-any.whl
  9.  
  10. !rm ./*.whl

4.3 准备部署模型

在进行模型服务化部署时，首先需要将推理模型转为用户服务化部署的模型。

首先运行下面的命令下载推理模型。

  1. os.chdir("/home/aistudio/PaddleOCR/deploy/pdserving/")
  2.  
  3. # 下载并解压 OCR 文本检测模型
  4. !wget https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_det_infer.tar -O ch_PP-OCRv2_det_infer.tar && tar -xf ch_PP-OCRv2_det_infer.tar && rm ch_PP-OCRv2_det_infer.tar
  5. # 下载并解压 OCR 文本识别模型
  6. !wget https://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_rec_infer.tar -O ch_PP-OCRv2_rec_infer.tar &&  tar -xf ch_PP-OCRv2_rec_infer.tar && rm ch_PP-OCRv2_rec_infer.tar

运行下面的命令进行模型转换。

  1. # 转换检测模型
  2. !python -m paddle_serving_client.convert --dirname ./ch_PP-OCRv2_det_infer/ \
  3.                                          --model_filename inference.pdmodel          \
  4.                                          --params_filename inference.pdiparams       \
  5.                                          --serving_server ./ppocrv2_det_serving/ \
  6.                                          --serving_client ./ppocrv2_det_client/
  7.  
  8. # 转换识别模型
  9. !python -m paddle_serving_client.convert --dirname ./ch_PP-OCRv2_rec_infer/ \
  10.                                          --model_filename inference.pdmodel          \
  11.                                          --params_filename inference.pdiparams       \
  12.                                          --serving_server ./ppocrv2_rec_serving/  \
  13.                                          --serving_client ./ppocrv2_rec_client/
  14.  
  15. # 查看文件夹
  16. !tree -h *_client *_serving

检测模型转换完成后，会在当前文件夹多出ppocrv2_det_mobile_serving 和ppocrv2_det_mobile_client的文件夹，具备如下格式：

  1. |- ppocrv2_det_mobile_serving/
  2.   |- __model__  
  3.   |- __params__
  4.   |- serving_server_conf.prototxt  
  5.   |- serving_server_conf.stream.prototxt
  6.  
  7. |- ppocrv2_det_mobile_client
  8.   |- serving_client_conf.prototxt  
  9.   |- serving_client_conf.stream.prototxt

识别模型同理。

4.4 Paddle Serving pipeline部署

注意： 将PaddleOCR/deploy/pdserving/config.yml文件中的两个model_config字段分别修改为ppocrv2_det_mobile_serving、ppocrv2_rec_mobile_serving，对应模型转换的文件夹。

  1. pdserving目录包含启动pipeline服务和发送预测请求的代码，包括：
  2. ```
  3. __init__.py
  4. config.yml            # 启动服务的配置文件
  5. ocr_reader.py         # OCR模型预处理和后处理的代码实现
  6. pipeline_http_client.py   # 发送预测请求的脚本
  7. web_service.py        # 启动服务端的脚本
  8. ```

4.4.1 启动服务

运行如下命令启动服务：

开启新的终端运行下列启动服务的命令

  1. ```
  2. # 启动服务，运行日志保存在web_serving_log.txt
  3. cd PaddleOCR/deploy/pdserving/
  4. nohup python web_service.py &>web_serving_log.txt &
  5. ```

成功启动服务后，web_serving_log.txt中会打印类似如下日志

4.4.2 发送服务请求：

!python pipeline_http_client.py

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/364209
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号