changdazhou 4 月之前
父節點
當前提交
79422e06bc
共有 1 個文件被更改,包括 3 次插入5 次删除
  1. 3 5
      paddlex/inference/pipelines/ocr/result.py

+ 3 - 5
paddlex/inference/pipelines/ocr/result.py

@@ -81,12 +81,10 @@ class OCRResult(BaseCVResult):
         img_right = np.ones((h, w, 3), dtype=np.uint8) * 255
         random.seed(0)
         draw_left = ImageDraw.Draw(img_left)
+        vis_font = SIMFANG_FONT
+        if self["vis_fonts"]:
+            vis_font = self["vis_fonts"][0]
         for idx, (box, txt) in enumerate(zip(boxes, txts)):
-            vis_font = (
-                self["vis_fonts"][idx]
-                if self["vis_fonts"][idx] is not None
-                else SIMFANG_FONT
-            )
             try:
                 color = (
                     random.randint(0, 255),