提交PaddleOCR车牌识别推理示例

PaddleOCR_infer_migraphx.py

+import os
+import sys
+import cv2
+import time
+import onnx
+import math
+import copy
+import migraphx
+import numpy as np
+import pyclipper
+from shapely.geometry import Polygon
+
+# PalldeOCR 检测模块 需要用到的图片预处理类
+class NormalizeImage(object):
+    def __init__(self, scale=None, mean=None, std=None, order='chw', **kwargs):
+        if isinstance(scale, str):
+            scale = eval(scale)
+        self.scale = np.float32(scale if scale is not None else 1.0 / 255.0)
+        mean = mean if mean is not None else [0.485, 0.456, 0.406]
+        std = std if std is not None else [0.229, 0.224, 0.225]
+
+        shape = (3, 1, 1) if order == 'chw' else (1, 1, 3)
+        self.mean = np.array(mean).reshape(shape).astype('float32')
+        self.std = np.array(std).reshape(shape).astype('float32')
+
+    def __call__(self, data):
+        img = data['image']
+        from PIL import Image
+        if isinstance(img, Image.Image):
+            img = np.array(img)
+
+        assert isinstance(img,
+                          np.ndarray), "invalid input 'img' in NormalizeImage"
+        data['image'] = (
+            img.astype('float32') * self.scale - self.mean) / self.std
+        return data
+
+
+class ToCHWImage(object):
+    def __init__(self, **kwargs):
+        pass
+
+    def __call__(self, data):
+        img = data['image']
+        from PIL import Image
+        if isinstance(img, Image.Image):
+            img = np.array(img)
+        data['image'] = img.transpose((2, 0, 1))
+        return data
+
+
+class KeepKeys(object):
+    def __init__(self, keep_keys, **kwargs):
+        self.keep_keys = keep_keys
+
+    def __call__(self, data):
+        data_list = []
+        for key in self.keep_keys:
+            data_list.append(data[key])
+        return data_list
+
+class DetResizeForTest(object):
+    
+    def __init__(self, **kwargs):
+        super(DetResizeForTest, self).__init__()
+        self.resize_type = 0
+        self.limit_side_len = kwargs['limit_side_len']
+        self.limit_type = kwargs.get('limit_type', 'min')
+
+    def __call__(self, data):
+        img = data['image']
+        src_h, src_w, _ = img.shape
+        img, [ratio_h, ratio_w] = self.resize_image_type0(img)
+        data['image'] = img
+        data['shape'] = np.array([src_h, src_w, ratio_h, ratio_w])
+        return data
+
+    def resize_image_type0(self, img):
+        limit_side_len = self.limit_side_len
+        h, w, _ = img.shape
+
+        if max(h, w) > limit_side_len:
+            if h > w:
+                ratio = float(limit_side_len) / h
+            else:
+                ratio = float(limit_side_len) / w
+        else:
+            ratio = 1.
+        resize_h = int(h * ratio)
+        resize_w = int(w * ratio)
+
+
+        resize_h = int(round(resize_h / 32) * 32)
+        resize_w = int(round(resize_w / 32) * 32)
+
+        try:
+            if int(resize_w) <= 0 or int(resize_h) <= 0:
+                return None, (None, None)
+            img = cv2.resize(img, (int(resize_w), int(resize_h)))
+        except:
+            print(img.shape, resize_w, resize_h)
+            sys.exit(0)
+        ratio_h = resize_h / float(h)
+        ratio_w = resize_w / float(w)
+        
+        return img, [ratio_h, ratio_w]
+
+
+# 检测结果后处理
+class DBPostProcess(object):
+
+    def __init__(self,
+                 thresh=0.3,
+                 box_thresh=0.7,
+                 max_candidates=1000,
+                 unclip_ratio=2.0,
+                 use_dilation=False,
+                 **kwargs):
+        self.thresh = thresh
+        self.box_thresh = box_thresh
+        self.max_candidates = max_candidates
+        self.unclip_ratio = unclip_ratio
+        self.min_size = 3
+        self.dilation_kernel = None if not use_dilation else np.array([[1, 1], [1, 1]])
+
+    def boxes_from_bitmap(self, pred, _bitmap, dest_width, dest_height):
+        bitmap = _bitmap
+        height, width = bitmap.shape
+
+        outs = cv2.findContours((bitmap * 255).astype(np.uint8), cv2.RETR_LIST,
+                                cv2.CHAIN_APPROX_SIMPLE)
+        if len(outs) == 3:
+            img, contours, _ = outs[0], outs[1], outs[2]
+        elif len(outs) == 2:
+            contours, _ = outs[0], outs[1]
+
+        num_contours = min(len(contours), self.max_candidates)
+
+        boxes = []
+        scores = []
+        for index in range(num_contours):
+            contour = contours[index]
+            points, sside = self.get_mini_boxes(contour)
+            if sside < self.min_size:
+                continue
+            points = np.array(points)
+            score = self.box_score_fast(pred, points.reshape(-1, 2))
+            if self.box_thresh > score:
+                continue
+
+            box = self.unclip(points).reshape(-1, 1, 2)
+            box, sside = self.get_mini_boxes(box)
+            if sside < self.min_size + 2:
+                continue
+            box = np.array(box)
+
+            box[:, 0] = np.clip(
+                np.round(box[:, 0] / width * dest_width), 0, dest_width)
+            box[:, 1] = np.clip(
+                np.round(box[:, 1] / height * dest_height), 0, dest_height)
+            boxes.append(box.astype(np.int16))
+            scores.append(score)
+            
+        return np.array(boxes, dtype=np.int16), scores
+
+    def unclip(self, box):
+        unclip_ratio = self.unclip_ratio
+        poly = Polygon(box)
+        distance = poly.area * unclip_ratio / poly.length
+        offset = pyclipper.PyclipperOffset()
+        offset.AddPath(box, pyclipper.JT_ROUND, pyclipper.ET_CLOSEDPOLYGON)
+        expanded = np.array(offset.Execute(distance))
+        return expanded
+
+    def get_mini_boxes(self, contour):
+        bounding_box = cv2.minAreaRect(contour)
+        points = sorted(list(cv2.boxPoints(bounding_box)), key=lambda x: x[0])
+
+        index_1, index_2, index_3, index_4 = 0, 1, 2, 3
+        if points[1][1] > points[0][1]:
+            index_1 = 0
+            index_4 = 1
+        else:
+            index_1 = 1
+            index_4 = 0
+        if points[3][1] > points[2][1]:
+            index_2 = 2
+            index_3 = 3
+        else:
+            index_2 = 3
+            index_3 = 2
+
+        box = [
+            points[index_1], points[index_2], points[index_3], points[index_4]
+        ]
+        return box, min(bounding_box[1])
+
+    def box_score_fast(self, bitmap, _box):
+        h, w = bitmap.shape[:2]
+        box = _box.copy()
+        xmin = np.clip(np.floor(box[:, 0].min()).astype(int), 0, w - 1)
+        xmax = np.clip(np.ceil(box[:, 0].max()).astype(int), 0, w - 1)
+        ymin = np.clip(np.floor(box[:, 1].min()).astype(int), 0, h - 1)
+        ymax = np.clip(np.ceil(box[:, 1].max()).astype(int), 0, h - 1)
+
+        mask = np.zeros((ymax - ymin + 1, xmax - xmin + 1), dtype=np.uint8)
+        box[:, 0] = box[:, 0] - xmin
+        box[:, 1] = box[:, 1] - ymin
+        cv2.fillPoly(mask, box.reshape(1, -1, 2).astype(np.int32), 1)
+        return cv2.mean(bitmap[ymin:ymax + 1, xmin:xmax + 1], mask)[0]
+
+    def __call__(self, outs_dict, shape_list):
+        pred = outs_dict
+        pred = pred[:, 0, :, :]
+        segmentation = pred > self.thresh
+        
+
+        boxes_batch = []
+        for batch_index in range(pred.shape[0]):
+            src_h, src_w, ratio_h, ratio_w = shape_list[batch_index]
+            if self.dilation_kernel is not None:
+                mask = cv2.dilate(
+                    np.array(segmentation[batch_index]).astype(np.uint8),
+                    self.dilation_kernel)
+            else:
+                mask = segmentation[batch_index]
+            boxes, scores = self.boxes_from_bitmap(pred[batch_index], mask,
+                                                   src_w, src_h)
+
+            boxes_batch.append({'points': boxes})
+        return boxes_batch
+
+# 识别结果解码过程
+class process_pred(object):
+    def __init__(self, character_dict_path=None, character_type='ch', use_space_char=False):
+        self.character_str = ''
+        with open(character_dict_path, 'rb') as fin:
+            lines = fin.readlines()
+            for line in lines:
+                line = line.decode('utf-8').strip('\n').strip('\r\n')
+                self.character_str += line
+        if use_space_char:
+            self.character_str += ' '
+        dict_character = list(self.character_str)
+
+        dict_character = self.add_special_char(dict_character)
+        self.dict = {}
+        for i, char in enumerate(dict_character):
+            self.dict[char] = i
+        self.character = dict_character
+
+    def add_special_char(self, dict_character):
+        dict_character = ['blank'] + dict_character
+        return dict_character
+
+    def decode(self, text_index, text_prob=None, is_remove_duplicate=False):
+        result_list = []
+        ignored_tokens = [0]
+        batch_size = len(text_index)
+        for batch_idx in range(batch_size):
+            char_list = []
+            conf_list = []
+            for idx in range(len(text_index[batch_idx])):
+                if text_index[batch_idx][idx] in ignored_tokens:
+                    continue
+                if is_remove_duplicate:
+                    if idx > 0 and text_index[batch_idx][idx - 1] == text_index[batch_idx][idx]:
+                        continue
+                char_list.append(self.character[int(text_index[batch_idx][idx])])
+                if text_prob is not None:
+                    conf_list.append(text_prob[batch_idx][idx])
+                else:
+                    conf_list.append(1)
+            text = ''.join(char_list)
+            result_list.append((text, np.mean(conf_list)))
+        return result_list
+
+    def __call__(self, preds, label=None):
+        if not isinstance(preds, np.ndarray):
+            preds = np.array(preds)
+        preds_idx = preds.argmax(axis=2)
+        preds_prob = preds.max(axis=2)
+        text = self.decode(preds_idx, preds_prob, is_remove_duplicate=True)
+        if label is None:
+            return text
+        label = self.decode(label)
+        return text, label
+
+# 检测＋识别推理代码
+class det_rec_functions(object):
+
+    def __init__(self, image):
+        self.img = image.copy()
+        self.det_file = './weights/db.onnx'
+        self.rec_file = './weights/svtr.onnx'
+
+        # 解析检测模型
+        detInput = {"x":[1,3,2496,2496]}
+        self.modelDet = migraphx.parse_onnx(self.det_file, map_input_dims=detInput)
+        self.inputName = self.modelDet.get_parameter_names()[0]
+        self.inputShape = self.modelDet.get_parameter_shapes()[self.inputName].lens()
+
+        print("DB inputName:{0} \nDB inputShape:{1}".format(self.inputName, self.inputShape))
+
+        # 模型编译
+        self.modelDet.compile(t=migraphx.get_target("gpu"), device_id=0)  # device_id: 设置GPU设备，默认为0号设备
+        print("Success to compile DB")
+
+        # 解析识别模型
+        recInput = {"x":[1,3,48,320]}
+        self.modelRec = migraphx.parse_onnx(self.rec_file, map_input_dims=recInput)
+        self.inputName = self.modelRec.get_parameter_names()[0]
+        self.inputShape = self.modelRec.get_parameter_shapes()[self.inputName].lens()
+
+        print("SVTR inputName:{0} \nSVTR inputShape:{1}".format(self.inputName, self.inputShape))
+
+        # 模型编译
+        self.modelRec.compile(t=migraphx.get_target("gpu"), device_id=0)  # device_id: 设置GPU设备，默认为0号设备
+        print("Success to compile SVTR")
+        
+        self.infer_before_process_op, self.det_re_process_op = self.get_process()
+        self.postprocess_op = process_pred('./weights/ppocr_keys_v1.txt', 'ch', True)
+
+    # 图片预处理过程
+    def transform(self, data, ops=None):
+        if ops is None:
+            ops = []
+        for op in ops:
+            data = op(data)
+            if data is None:
+                return None
+        return data
+
+    def create_operators(self, op_param_list, global_config=None):
+        assert isinstance(op_param_list, list), ('operator config should be a list')
+        ops = []
+        for operator in op_param_list:
+            assert isinstance(operator,
+                              dict) and len(operator) == 1, "yaml format error"
+            op_name = list(operator)[0]
+            param = {} if operator[op_name] is None else operator[op_name]
+            if global_config is not None:
+                param.update(global_config)
+            op = eval(op_name)(**param)
+            ops.append(op)
+        return ops
+
+    # 检测框的后处理
+    def order_points_clockwise(self, pts):
+        
+        xSorted = pts[np.argsort(pts[:, 0]), :]
+        leftMost = xSorted[:2, :]
+        rightMost = xSorted[2:, :]
+        leftMost = leftMost[np.argsort(leftMost[:, 1]), :]
+        (tl, bl) = leftMost
+
+        rightMost = rightMost[np.argsort(rightMost[:, 1]), :]
+        (tr, br) = rightMost
+
+        rect = np.array([tl, tr, br, bl], dtype="float32")
+        return rect
+
+    def clip_det_res(self, points, img_height, img_width):
+        for pno in range(points.shape[0]):
+            points[pno, 0] = int(min(max(points[pno, 0], 0), img_width - 1))
+            points[pno, 1] = int(min(max(points[pno, 1], 0), img_height - 1))
+        return points
+
+    def filter_tag_det_res(self, dt_boxes, image_shape):
+        img_height, img_width = image_shape[0:2]
+        dt_boxes_new = []
+        for box in dt_boxes:
+            box = self.order_points_clockwise(box)
+            box = self.clip_det_res(box, img_height, img_width)
+            rect_width = int(np.linalg.norm(box[0] - box[1]))
+            rect_height = int(np.linalg.norm(box[0] - box[3]))
+            if rect_width <= 3 or rect_height <= 3:
+                continue
+            dt_boxes_new.append(box)
+        dt_boxes = np.array(dt_boxes_new)
+        return dt_boxes
+
+    # 定义图片前处理过程，和检测结果后处理过程
+    def get_process(self):
+        det_db_thresh = 0.3
+        det_db_box_thresh = 0.5
+        max_candidates = 2000
+        unclip_ratio = 1.6
+        use_dilation = True
+
+        pre_process_list = [{
+            'DetResizeForTest': {
+                'limit_side_len': 2500,
+                'limit_type': 'max'
+            }
+        }, {
+            'NormalizeImage': {
+                'std': [0.229, 0.224, 0.225],
+                'mean': [0.485, 0.456, 0.406],
+                'scale': '1./255.',
+                'order': 'hwc'
+            }
+        }, {
+            'ToCHWImage': None
+        }, {
+            'KeepKeys': {
+                'keep_keys': ['image', 'shape']
+            }
+        }]
+
+        infer_before_process_op = self.create_operators(pre_process_list)
+        det_re_process_op = DBPostProcess(det_db_thresh, det_db_box_thresh, max_candidates, unclip_ratio, use_dilation)
+        return infer_before_process_op, det_re_process_op
+
+    def sorted_boxes(self, dt_boxes):
+        num_boxes = dt_boxes.shape[0]
+        sorted_boxes = sorted(dt_boxes, key=lambda x: (x[0][1], x[0][0]))
+        _boxes = list(sorted_boxes)
+
+        for i in range(num_boxes - 1):
+            if abs(_boxes[i + 1][0][1] - _boxes[i][0][1]) < 10 and \
+                    (_boxes[i + 1][0][0] < _boxes[i][0][0]):
+                tmp = _boxes[i]
+                _boxes[i] = _boxes[i + 1]
+                _boxes[i + 1] = tmp
+        return _boxes
+
+    # 图像输入预处理
+    def resize_norm_img(self, img, max_wh_ratio):
+        imgC, imgH, imgW = [int(v) for v in "3, 48, 320".split(",")]
+        assert imgC == img.shape[2]
+        imgW = int((48 * max_wh_ratio))
+        h, w = img.shape[:2]
+        ratio = w / float(h)
+        if math.ceil(imgH * ratio) > imgW:
+           resized_w = imgW
+        else:
+           resized_w = int(math.ceil(imgH * ratio))
+        resized_image = cv2.resize(img, (resized_w, imgH))
+        
+        resized_image = resized_image.astype('float32')
+        resized_image = resized_image.transpose((2, 0, 1)) / 255
+        resized_image -= 0.5
+        resized_image /= 0.5
+        padding_im = np.zeros((imgC, imgH, imgW), dtype=np.float32)
+        padding_im[:, :, 0:resized_w] = resized_image
+        return padding_im
+
+    # 推理检测图片中的部分
+    def get_boxes(self):
+        img_ori = self.img
+        img_part = img_ori.copy()
+        data_part = {'image': img_part}
+        data_part = self.transform(data_part, self.infer_before_process_op)
+        img_part, shape_part_list = data_part
+        img_part = np.expand_dims(img_part, axis=0)
+        shape_part_list = np.expand_dims(shape_part_list, axis=0)
+
+        # migraphx推理
+        inputOfReshape = img_part.shape
+        inputShapeMap={self.inputName:[1,3,inputOfReshape[2],inputOfReshape[3]]}
+        migraphx.reshape2(self.modelDet, inputShapeMap)
+        resultDets = self.modelDet.run({self.modelDet.get_parameter_names()[0]: migraphx.argument(img_part)})
+        # 获取第一个输出节点的数据,migraphx.argument类型
+        resultDet = resultDets[0]
+        outs_part = np.array(resultDet)
+
+        post_res_part = self.det_re_process_op(outs_part, shape_part_list)
+        dt_boxes_part = post_res_part[0]['points']
+        dt_boxes_part = self.filter_tag_det_res(dt_boxes_part, img_ori.shape)
+        dt_boxes_part = self.sorted_boxes(dt_boxes_part)
+        
+        return dt_boxes_part
+
+    # 根据bounding box得到单元格图片
+    def get_rotate_crop_image(self, img, points):
+        img_crop_width = int(
+            max(
+                np.linalg.norm(points[0] - points[1]),
+                np.linalg.norm(points[2] - points[3])))
+        img_crop_height = int(
+            max(
+                np.linalg.norm(points[0] - points[3]),
+                np.linalg.norm(points[1] - points[2])))
+        pts_std = np.float32([[0, 0], [img_crop_width, 0],
+                              [img_crop_width, img_crop_height],
+                              [0, img_crop_height]])
+        M = cv2.getPerspectiveTransform(points, pts_std)
+        dst_img = cv2.warpPerspective(
+            img,
+            M, (img_crop_width, img_crop_height),
+            borderMode=cv2.BORDER_REPLICATE,
+            flags=cv2.INTER_CUBIC)
+        dst_img_height, dst_img_width = dst_img.shape[0:2]
+        if dst_img_height * 1.0 / dst_img_width >= 1.5:
+            dst_img = np.rot90(dst_img)
+        return dst_img
+
+    # 单张图片推理
+    def get_img_res(self, img, process_op):
+        h, w = img.shape[:2]
+        img = self.resize_norm_img(img, w * 1.0 / h)
+        img = img[np.newaxis, :]
+        
+        # migraphx推理
+        inputOfReshape = img.shape
+        inputShapeMap={self.inputName:[1,3,inputOfReshape[2],inputOfReshape[3]]}
+        migraphx.reshape2(self.modelRec, inputShapeMap)
+        
+        results = self.modelRec.run({self.modelRec.get_parameter_names()[0]: migraphx.argument(img)})
+        # 获取第一个输出节点的数据,migraphx.argument类型
+        result = results[0]
+        outs = np.array(result)
+
+        result = process_op(outs)
+        return result
+
+    def recognition_img(self, dt_boxes):
+        img_ori = self.img
+        img = img_ori.copy()
+        # 根据bndbox得到小图片
+        img_list = []
+        for box in dt_boxes:
+            tmp_box = copy.deepcopy(box)
+            img_crop = self.get_rotate_crop_image(img, tmp_box)
+            
+            img_list.append(img_crop)
+
+        # 识别小图片
+        results = []
+        results_info = []
+        for pic in img_list:
+            res = self.get_img_res(pic, self.postprocess_op)
+            results_info.append(res)
+        return results_info
+
+if __name__=='__main__':
+  
+    image = cv2.imread('./images/vlpr.jpg')
+    start = time.time()
+    ocr_sys = det_rec_functions(image)
+    dt_boxes = ocr_sys.get_boxes()
+    results_info = ocr_sys.recognition_img(dt_boxes)
+    
+    print('net forward time: {:.4f}'.format(time.time() - start))
+    print("############# OCR Results #############")
+    print(results_info)
\ No newline at end of file

README.md

-# PaddleOCR_MIGraphX
+# PaddleOCR车牌识别
 
-本示例基于MIGraphX构建PaddleOCR推理示例
+## 模型介绍
\ No newline at end of file
+
+车牌识别(Vehicle License Plate Recognition，VLPR) 是计算机视频图像识别技术在车辆牌照识别中的一种应用。车牌识别技术要求能够将运动中的汽车牌照从复杂背景中提取并识别出来，在高速公路车辆管理，停车场管理和城市交通中得到广泛应用。PaddleOCR车牌识别包括本文检测和文本识别两部分内容，其中使用DBnet作为文本检测模型，SVTR作为文本识别模型。
+
+## 模型结构
+
+DBnet是一种基于分割的文本检测方法，相比传统分割方法需要设定固定阈值，该模型将二值化操作插入到分割网络中进行联合优化，通过网络学习可以自适应的预测图像中每一个像素点的阈值，能够在像素水平很好的检测自然场景下不同形状的文字。SVTR是一种端到端的文本识别模型，通过单个视觉模型就可以一站式解决特征提取和文本转录两个任务，同时也保证了更快的推理速度。百度PaddleOCR开源项目提供了车牌识别的预训练模型，本示例使用PaddleOCR提供的蓝绿黄牌识别模型进行推理，车牌识别过程：输入->图像预处理->文字检测->文本识别->输出。
+
+## 推理
+
+### 环境配置
+
+在[光源](https://www.sourcefind.cn/#/image/dcu/custom)可拉取用于推理的docker镜像，PaddleOCR车牌识别模型推理推荐的镜像如下：
+
+```
+docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:ort_dcu_1.14.0_migraphx2.5.2_dtk22.10.1
+```
+
+在[光合开发者社区](https://cancon.hpccube.com:65024/4/main/)可下载MIGraphX安装包，python依赖安装：
+
+```
+pip install -r requirements.txt
+```
+
+注意：PaddleOCR车牌识别使用动态shape推理，需要MIGraphX版本>=2.5.2。
+
+### 运行示例
+
+PaddleOCR车牌识别的推理示例程序是PaddleOCR_infer_migraphx.py，使用如下命令运行该推理示例：
+
+```
+# 设置动态shape模式
+export MIGRAPHX_DYNAMIC_SHAPE=1
+
+# 运行示例
+python PaddleOCR_infer_migraphx.py
+```
+
+车牌识别结果为：
+
+```
+[[('皖AD19906', 0.98606485)]]
+```
+
+## 历史版本
+
+​		https://developer.hpccube.com/codes/modelzoo/paddleocr_migraphx
+
+## 参考
+
+​		https://github.com/PaddlePaddle/PaddleOCR

requirements.txt

+shapely
+pyclipper
+numpy
+opencv-contrib-python==4.6.0.66
+pillow
\ No newline at end of file