mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Enhancement] update stitch_boxes_into_lines (#1824)

pull/1840/head
cherryjm 2023-04-03 17:35:21 +08:00 committed by GitHub
parent 1c91a9820a
commit 4842599191
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mmocr/utils/bbox_utils.py
View File

@ -178,14 +178,15 @@ def stitch_boxes_into_lines(boxes, max_x_dist=10, min_y_overlap_ratio=0.8):
lines = []
line_idx = 0
lines.append([line[0]])
rightmost = np.max(x_sorted_boxes[line[0]]['box'][::2])
for k in range(1, len(line)):
curr_box = x_sorted_boxes[line[k]]
prev_box = x_sorted_boxes[line[k - 1]]
dist = np.min(curr_box['box'][::2]) - np.max(prev_box['box'][::2])
dist = np.min(curr_box['box'][::2]) - rightmost
if dist > max_x_dist:
line_idx += 1
lines.append([])
lines[line_idx].append(line[k])
rightmost = max(rightmost, np.max(curr_box['box'][::2]))
# Get merged boxes
for box_group in lines: