From 4c1e80b7231a81e898b2bbaf1df07cce136de38f Mon Sep 17 00:00:00 2001
From: DefTruth <31974251+DefTruth@users.noreply.github.com>
Date: Fri, 7 Jul 2023 17:43:08 +0800
Subject: [PATCH] [Bug Fix] fixed ocr visualize error (#2090)

---
 fastdeploy/vision/visualize/ocr.cc | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/fastdeploy/vision/visualize/ocr.cc b/fastdeploy/vision/visualize/ocr.cc
index 5d0bb9e1b..e35be9135 100644
--- a/fastdeploy/vision/visualize/ocr.cc
+++ b/fastdeploy/vision/visualize/ocr.cc
@@ -20,10 +20,14 @@ namespace vision {
 cv::Mat VisOcr(const cv::Mat& im, const OCRResult& ocr_result,
                const float score_threshold) {
   auto vis_im = im.clone();
+  bool have_score = 
+    (ocr_result.boxes.size() == ocr_result.rec_scores.size());
 
   for (int n = 0; n < ocr_result.boxes.size(); n++) {
-    if (ocr_result.rec_scores[n] < score_threshold) {
-      continue;
+    if (have_score) {
+      if (ocr_result.rec_scores[n] < score_threshold) {
+        continue;
+      }
     }
     cv::Point rook_points[4];