|
import numpy as np |
|
import cv2 |
|
import os |
|
import json |
|
from functools import cmp_to_key |
|
|
|
def resize_box(box, ori_img_shape, nomarlized_img_shape=(1000, 1000, 3)): |
|
""" |
|
box: [x0, y0, x1, y1], |
|
ori_img_shape: 形如(1560, 1103, 3) |
|
""" |
|
|
|
height_ratio = nomarlized_img_shape[0] / ori_img_shape[0] |
|
width_ratio = nomarlized_img_shape[1] / ori_img_shape[1] |
|
|
|
x0, y0, x1, y1 = box |
|
norm_x0, norm_x1 = round(x0 * width_ratio), round(x1 * width_ratio) |
|
norm_y0, norm_y1 = round(y0 * height_ratio), round(y1 * height_ratio) |
|
|
|
return [norm_x0, norm_y0, norm_x1, norm_y1] |
|
|
|
def tblr_reading_order_detector(tuple_list): |
|
"""rule: top-to-bottom, left-to-right |
|
|
|
tuple: (word_text, word_bbox, normed_word_bbox) |
|
|
|
return: sorted_tuple_list |
|
""" |
|
|
|
def sort_cmp_fn(word_box1, word_box2): |
|
""" |
|
sorted function的排序的2个元素的比较准则。 |
|
1. 比较box1和box2的y坐标，如果二者的高重合度达到了二者的50%，则位于同一行，否则位于不同行。 |
|
2. 如果位于同一行，那么比较二者的x0，如果box1_x0 < box2_x0，则返回-1，表示box_1<box_2，否则返回0（表示相等）或者1（box1>box2）。 |
|
3. 如果不位于同一行，那么比较二者的y0，如果box1_y0 < box2_y0，则返回-1，否则返回0或者1. |
|
""" |
|
|
|
x0, y0, x1, y1 = word_box1[1] |
|
x0_, y0_, x1_, y1_ = word_box2[1] |
|
|
|
|
|
if y0 < y0_: |
|
return -1 |
|
elif y0 > y0_: |
|
return 1 |
|
elif y0 == y0_: |
|
if x0 <= x0_: |
|
return -1 |
|
elif x0 > x0_: |
|
return 1 |
|
|
|
sorted_tuple_list = sorted(tuple_list, key=cmp_to_key(sort_cmp_fn)) |
|
|
|
|
|
return sorted_tuple_list |
|
|
|
|
