File size: 1,827 Bytes
41404b8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
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))
# print(sorted_word_box_list)
return sorted_tuple_list
|