Upload 6 files

ocr/__init__.py

@@ -0,0 +1,5 @@
+# -*- coding: utf-8 -*-
+# time: 2022/10/17 13:03
+# file: __init__.py.py
+# author: fengchenggang
+# email: chenggang.feng@ashermed.com

ocr/ocr.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# time: 2022/10/17 13:04
+# file: ocr.py
+
+
+from paddleocr import PaddleOCR
+from ocr_utils import draw_ocr, draw_ocr_box_txt
+
+ocr = PaddleOCR(lang='ch',use_angle_cls=True)
+
+
+def detect(image):
+    """
+    文本检测
+    args:
+        image(array): numpy格式的图片 'RGB'
+    return(array):
+        检测后的图片 numpy格式 'RGB'
+    """
+    result = ocr.ocr(image, rec=False)
+    im_show = draw_ocr(image, result)
+
+    return im_show
+
+
+def recognize(image, output_mode=0):
+    """
+    文本识别
+    args:
+        image(array): numpy格式的图片 'RGB'
+        output_mode(int): 图片输出模式 [0|1]
+    return(array):
+        识别后的图片 numpy格式 'RGB'
+    """
+    result = ocr.ocr(image)
+    boxes = [line[0] for line in result]
+    txts = [line[1][0] for line in result]
+    scores = [line[1][1] for line in result]
+    if output_mode == 0:
+        im_show = draw_ocr_box_txt(image, boxes, txts, scores)
+    elif output_mode == 1:
+        im_show = draw_ocr(image, boxes, txts, scores)
+
+    return im_show
+

ocr/utils.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+# time: 2022/10/17 13:04
+# file: utils.py
+
+
+import cv2
+import numpy as np
+
+
+def bytes_to_numpy(image_bytes, channels='BGR'):
+    """
+    图片格式转换 bytes -> numpy
+    args:
+        image_bytes(str): 图片的字节流
+        channels(str): 图片的格式 ['BGR'|'RGB']
+    return(array):
+        转换后的图片
+    """
+    _image_np = np.frombuffer(image_bytes, dtype=np.uint8)
+    image_np = cv2.imdecode(_image_np, cv2.IMREAD_COLOR)
+    if channels == 'BGR':
+        return image_np
+    elif channels == 'RGB':
+        image_np = cv2.cvtColor(image_np, cv2.COLOR_BGR2RGB)
+        return image_np
+
+
+def numpy_to_pic(image_path, image_np, channels='BGR'):
+    """
+    保存图片
+    args:
+        image_path(str): 图片路径
+        image_np(array): numpy格式的图片数据
+        channels(str): 图片的格式 ['BGR'|'RGB']
+    """
+    if channels == 'BGR':
+        cv2.imwrite(image_path, image_np)
+    elif channels == 'RGB':
+        image_np = cv2.cvtColor(image_np, cv2.COLOR_RGB2BGR)
+        cv2.imwrite(image_path, image_np)