bbox 四个顶点排序

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 的是左下。下图左边对应的就是这种一般性的解决方案,四个顶点分别位于中心点的左上、右上、右下、左下四个位置。但是如果遇到右边这种情况,有两个顶点位于中心点的右上、有两个顶点位于中心点的左下,就不能按上述方案解决了!

image-20210723163039703

这里提供另一种解决思路:先计算矩形框对角线的交点坐标 ( 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)下图右边的情况,有两个点横坐标是相同的,有两个点纵坐标是相同的。

image-20210723164600031

所以需要对“中”“右”这两种特例做特殊处理:(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)
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值