Skip to content

Commit

Permalink
fix(rapidocr_onnxruntime): ensure 4 point for char box rec (#294)
Browse files Browse the repository at this point in the history
  • Loading branch information
Joker1212 authored Dec 12, 2024
1 parent 0d70e86 commit fef40a5
Showing 1 changed file with 14 additions and 7 deletions.
21 changes: 14 additions & 7 deletions python/rapidocr_paddle/cal_rec_boxes/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -137,8 +137,8 @@ def adjust_box_overlap(
distance = abs(cur[1][0] - nxt[0][0])
cur[1][0] -= distance / 2
cur[2][0] -= distance / 2
nxt[0][0] += distance / 2
nxt[3][0] += distance / 2
nxt[0][0] += distance - distance / 2
nxt[3][0] += distance - distance / 2
return word_box_list

def reverse_rotate_crop_image(
Expand Down Expand Up @@ -218,6 +218,13 @@ def s_rotate(angle, valuex, valuey, pointx, pointy):
@staticmethod
def order_points(box: List[List[int]]) -> List[List[int]]:
"""矩形框顺序排列"""
def convert_to_1x2(p):
if p.shape == (2,):
return p.reshape((1, 2))
elif p.shape == (1, 2):
return p
else:
return p[:1, :]
box = np.array(box).reshape((-1, 2))
center_x, center_y = np.mean(box[:, 0]), np.mean(box[:, 1])
if np.any(box[:, 0] == center_x) and np.any(
Expand Down Expand Up @@ -261,9 +268,9 @@ def order_points(box: List[List[int]]) -> List[List[int]]:
p23[np.where(p23[:, 1] == np.min(p23[:, 1]))],
p23[np.where(p23[:, 1] == np.max(p23[:, 1]))],
)
# 解决单字矩形框重叠导致多个相同框的情况
p1 = p1[:1, :]
p2 = p2[:1, :]
p3 = p3[:1, :]
p4 = p4[:1, :]
# 解决单字切割后横坐标完全相同的shape错误
p1 = convert_to_1x2(p1)
p2 = convert_to_1x2(p2)
p3 = convert_to_1x2(p3)
p4 = convert_to_1x2(p4)
return np.array([p1, p2, p3, p4]).reshape((-1, 2)).tolist()

0 comments on commit fef40a5

Please sign in to comment.