From 5052fe1ed9bcc5b2412a71cb3e810e4d18f15857 Mon Sep 17 00:00:00 2001 From: SWHL Date: Sun, 8 Dec 2024 10:48:49 +0800 Subject: [PATCH] fix: fixed the error of visulizing single word --- docs/install_usage/api/VisRes.md | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/docs/install_usage/api/VisRes.md b/docs/install_usage/api/VisRes.md index 73be74f8..11c8ee12 100644 --- a/docs/install_usage/api/VisRes.md +++ b/docs/install_usage/api/VisRes.md @@ -54,10 +54,9 @@ def __call__( image_path = "tests/test_files/ch_en_num.jpg" img = cv2.imread(image_path) - result, elapse_list = engine(img) - boxes, txts, scores = list(zip(*result)) - vis_img = vis(img, boxes, txts, scores, font_path) - cv2.imwrite("vis.png", vis_img) + result, elapse_list = engine(img, return_word_box=True) + + (boxes, txts, scores, words_boxes, words) = list(zip(*result)) words_boxes = sum(words_boxes, []) words_all = sum(words, [])