feat(rapidocr): 按照文本行返回单字坐标 (#466)

* 单字左边返回

按照文本行返回单字坐标,保持原先行格式

* Update main.py

get_origin_points 更改为:_get_origin_points
This commit is contained in:
qianliyx authored and GitHub committed 2025-06-18 07:59:22 +08:00
1 parent fb441e460a
commit ea04c30d40
2 files changed
+17 -10

No files matched your search

+1 -1
View File
@@ -47,7 +47,7 @@ class CalRecBoxes:
word_box_list = self.reverse_rotate_crop_image(
copy.deepcopy(box), word_box_list, direction
)
word_results.extend(
word_results.append(
list(zip(word_box_content_list, conf_list, word_box_list))
)
+16 -9
View File
@@ -123,16 +123,23 @@ class RapidOCR:
img, det_res.boxes, rec_res, self.return_single_char_box
)
origin_words = []
for one_word in rec_res.word_results:
one_word_points = one_word[2]
if one_word_points is None:
continue
for one_word_list in rec_res.word_results:
origin_words_item = []
for one_word in one_word_list:
one_word_points = one_word[2]
if one_word_points is None:
continue
origin_words_points = self._get_origin_points(
[one_word_points], op_record, raw_h, raw_w
)
origin_words_points = origin_words_points.astype(np.int32).tolist()[0]
origin_words.append((one_word[0], one_word[1], origin_words_points))
origin_words_points = self._get_origin_points(
[one_word_points], op_record, raw_h, raw_w
)
origin_words_points = origin_words_points.astype(np.int32).tolist()[0]
origin_words_item.append(
(one_word[0], one_word[1], origin_words_points)
)
if origin_words_item:
origin_words.append(tuple(origin_words_item))
rec_res.word_results = tuple(origin_words)
if det_res.boxes is not None: