docs: update vis script of return_word_box

RapidAI · Jan 19, 2025 · be6fc41 · be6fc41
1 parent 13c60ab
commit be6fc41
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/docs/install_usage/api/VisRes.md b/docs/install_usage/api/VisRes.md
@@ -56,11 +56,12 @@ def __call__(
 
     result, elapse_list = engine(img, return_word_box=True)
 
-    (boxes, txts, scores, words_boxes, words) = list(zip(*result))
+    (boxes, txts, scores, words_boxes, words, words_scores) = list(zip(*result))
+    font_path = "resources/fonts/FZYTK.TTF"
 
     words_boxes = sum(words_boxes, [])
     words_all = sum(words, [])
-    words_scores = [1.0] * len(words_boxes)
+    words_scores = sum(words_scores, [])
     vis_img = vis(img, words_boxes, words_all, words_scores, font_path)
     cv2.imwrite("vis_single.png", vis_img)
     ```