1、问题
已知矩形框(或者任意形状框,但必须是凸包)的四个顶点坐标,将其按照 “左上→右上→右下→左下” 的顺序排序。
2、思路
一般性的解决方案:先计算矩形框对角线的交点坐标 ( C x , C y ) (C_x, C_y) (Cx,Cy),则: x < C x , y < C y x<C_x, y<C_y x<Cx,y<Cy 的是左上, x > C x , y < C y x>C_x, y<C_y x>Cx,y<Cy 的是右上, x > C x , y > C y x>C_x, y>C_y x>Cx,y>Cy 的是右下, x < C x , y > C y x<C_x, y>C_y x<Cx,y>Cy 的是左下。下图左边对应的就是这种一般性的解决方案,四个顶点分别位于中心点的左上、右上、右下、左下四个位置。但是如果遇到右边这种情况,有两个顶点位于中心点的右上、有两个顶点位于中心点的左下,就不能按上述方案解决了!
这里提供另一种解决思路:先计算矩形框对角线的交点坐标 ( C x , C y ) (C_x, C_y) (Cx,Cy), x < C x x<C_x x<Cx 的两个点中 y y y 值较小的是左上,较大的是左下; x > C x x>C_x x>Cx 的两个点中 y y y 值较小的是右上,较大的是右下!这种思路先划分左右,再划分上下,但是也会存在一些特例:(1)下图中间的情况,有两个点横坐标是相同的,此时就无法将四个点分成左右两组;(2)下图右边的情况,有两个点横坐标是相同的,有两个点纵坐标是相同的。
所以需要对“中”“右”这两种特例做特殊处理:(1)横坐标相同时,按照“先上下后左右”的顺序处理,也就是 y < C y y<C_y y<Cy 的两个点中 x x x 较小的是左上,较大的是右上, y > C y y>C_y y>Cy 的两个点中 x x x 较小的是左下,较大的是右下;(2)菱形的情况“先左右后上下”、“先上下后左右”都无法处理,那就按照 x x x 最小的是左上、最大的是右下, y y y 最小的是右上、最大的是左下。
3、代码
def order_points(box):
'''矩形框顺序排列
:param box: numpy.array, shape=(4, 2)
:return:
'''
center_x, center_y = np.mean(box[:, 0]), np.mean(box[:, 1])
if np.any(box[:, 0] == center_x) and np.any(box[:, 1] == center_y): # 有两点横坐标相等,有两点纵坐标相等,菱形
p1 = box[np.where(box[:, 0] == np.min(box[:, 0]))]
p2 = box[np.where(box[:, 1] == np.min(box[:, 1]))]
p3 = box[np.where(box[:, 0] == np.max(box[:, 0]))]
p4 = box[np.where(box[:, 1] == np.max(box[:, 1]))]
elif np.any(box[:, 0] == center_x) and np.all(box[:, 1] != center_y): # 只有两点横坐标相等,先上下再左右
p12, p34 = box[np.where(box[:, 1] < center_y)], box[np.where(box[:, 1] > center_y)]
p1, p2 = p12[np.where(p12[:, 0] == np.min(p12[:, 0]))], p12[np.where(p12[:, 0] == np.max(p12[:, 0]))]
p3, p4 = p34[np.where(p34[:, 0] == np.max(p34[:, 0]))], p34[np.where(p34[:, 0] == np.min(p34[:, 0]))]
else: # 只有两点纵坐标相等,或者是没有相等的,先左右再上下
p14, p23 = box[np.where(box[:, 0] < center_x)], box[np.where(box[:, 0] > center_x)]
p1, p4 = p14[np.where(p14[:, 1] == np.min(p14[:, 1]))], p14[np.where(p14[:, 1] == np.max(p14[:, 1]))]
p2, p3 = p23[np.where(p23[:, 1] == np.min(p23[:, 1]))], p23[np.where(p23[:, 1] == np.max(p23[:, 1]))]
return np.array([p1, p2, p3, p4]).reshape(-1, 2)