更新OCR版本，Bata版，还不能上线

util/util.py

@@ -12,9 +12,10 @@ def get_default_datetime():
     return datetime.now().strftime('%Y-%m-%d %H:%M:%S')
 
 
-def get_ocr_layout(ocr, img_path):
+def get_ocr_layout(ocr, img_path, is_screenshot=False):
     """
     获取ocr识别的结果，转为合适的layout形式
+    :param is_screenshot: 是否是截图
     :param ocr: ocr模型
     :param img_path: 图片本地路径
     :return:
@@ -36,18 +37,18 @@ def get_ocr_layout(ocr, img_path):
         return True
 
     layout = []
-    ocr_result = ocr.ocr(img_path, cls=False)
-    ocr_result = ocr_result[0]
+    ocr_result = ocr.predict(input=img_path, use_doc_orientation_classify=not is_screenshot, use_doc_unwarping=not is_screenshot)
+    ocr_result = next(ocr_result)
     if not ocr_result:
-        return layout
-    for segment in ocr_result:
-        box = segment[0]
+        return layout, "0"
+    angle = ocr_result.get("doc_preprocessor_res", {}).get("angle", "0")
+    for i in range(len(ocr_result.get('rec_texts'))):
+        box = ocr_result.get("rec_polys")[i].tolist()
         box = _get_box(box)
         if not _normal_box(box):
             continue
-        text = segment[1][0]
-        layout.append((box, text))
-    return layout
+        layout.append((box, ocr_result.get("rec_texts")[i]))
+    return layout, str(angle)
 
 
 def delete_temp_file(temp_files):