mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-10 21:10:55 +08:00
feat(rapidocr): 按照文本行返回单字坐标 (#466)
* 单字左边返回 按照文本行返回单字坐标,保持原先行格式 * Update main.py get_origin_points 更改为:_get_origin_points
This commit is contained in:
1 parent
fb441e460a
commit
ea04c30d40
2 files changed
+17
-10
No files matched your search
@@ -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
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user