从上到下，从左到右对页面（2D 平面）上的框进行排序。 （Python）

Question

我正在尝试对 OCR 引擎的结果进行排序。 来自引擎的边界框是随机的，没有任何特定的排序。 我想从左到右，从上到下对盒子进行排序。 哪个是同一行（行）中的框应该从左到右排序，而不是 go 到下行（行）并从左到右排序，依此类推。

我有每个矩形（框）的所有 4 个点

这是一些示例图像。

示例图片 1

示例图像 2

示例图像 3

Answer 1

You can use sorted function, which is build-in Python function, as below (this code supposes that rects are stored as tuples of (x, y, width, height) or (x1, y1, x2, y2) :

from operator import itemgetter, attrgetter
rects = [(8, 10, 10, 10), (0, 5, 10, 10), (0, 0, 10, 10), (1, 10, 10, 10)]

rects = sorted(rects, key=itemgetter(1,0))

结果：

[(0, 0, 10, 10), (0, 5, 10, 10), (1, 10, 10, 10), (8, 10, 10, 10)]

Answer 2

这是一个提议：

• 按框中心纵坐标排序；
• 如果它们的垂直重叠大于第一个框高度的 50%，则在排序列表中相邻的两个框被视为“在同一行”。

我们将使用 python 的sorted和more_itertools.split_when来排序，然后根据这些标准进行分组。

from more_itertools import split_when

def get_y_center(b):
    up, down= b[0], b[2]
    return (up + down) / 2

def not_vertically_overlapping(b1, b2):
    up1, down1 = b1[0], b1[2]
    up2, down2 = b2[0], b2[2]
    return down1 < up2 or (down1 - up2) < (up2 - up1) 

def groupbyrow(boxes):
    sorted_boxes = sorted(boxes, key=get_y_center)
    return list(split_when(sorted_boxes, not_vertically_overlapping))

测试：

import random
import matplotlib.pyplot as plt
import matplotlib.colors as mcolors

# GENERATE RANDOM BOXES

centers = [(random.gauss(i, 5),random.gauss(j,0.25)) for i in range(0,40,10) for j in range(0,40,2)]
random.shuffle(centers)
boxes = [(y-(h:=random.gauss(1,0.25)),x-(w:=random.gauss(5,2)),y+h,x+w) for x,y in centers]

# GROUP BOXES BY ROW

rows = groupbyrow(boxes)

# DRAW BOXES WITH ONE COLOUR PER ROW

def draw_box(box, colour):
    u,l,d,r = box
    xs = [l, r, r, l, l]
    ys = [u, u, d, d, u]
    plt.plot(xs, ys, colour)

colours = list(mcolors.CSS4_COLORS.keys())

for i,row in enumerate(rows):
    for box in row:
        draw_box(box, colours[i])

plt.gca().set_aspect('equal')
plt.show()

Answer 3

您能否总结一下您正在考虑对盒子进行分类的规则？ 您想在 Zigzag 订单扫描中对盒子进行排序吗？ 无论如何，如果您想应用特定规则，您可以将自定义的比较器 function 传递给排序的 function。