【code】四坐标点位排序

最新推荐文章于 2024-07-03 08:59:08 发布

深度学习视觉

最新推荐文章于 2024-07-03 08:59:08 发布

阅读量957

点赞数

分类专栏： code 文章标签： python

本文链接：https://blog.csdn.net/lucky_kai/article/details/105218830

版权

code 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

sortCoordinate

排序坐标，使得坐标符合(x_min,y_min),(x_max,y_min),(x_max,y_max),(x_min,y_max)

def sortCoordinate(box):

    '''
    :param box: x,y坐标共4点8值;[x1, y1, x2, y2, x3, y3, x4, y4]
    x1y1->x2y2->x3y3-x4y4
    top_left->top_right->bottom_right->bottom_letft
    (x_min,y_min),(x_max,y_min),(x_max,y_max),(x_min,y_max)
    ------------------
    | *********      |
    |  *********     |
    |   *********    |
    |    *********   |
    |     *********  |
    ------------------
    :return: [[x1, y1], [x2, y2], [x3, y3], [x4, y4]]
    '''

    x1, y1, x2, y2, x3, y3, x4, y4 = box[:8]
    newBox = [[x1, y1], [x2, y2], [x3, y3], [x4, y4]]
    
    # 按第一维度正序排序[x_min,x_min,x_max,x_max]
    newBox = sorted(newBox, key=lambda x: x[0],reverse=False) 
    # 排序后，前两个是左边的上下两个坐标点x_min，按第二维排序，使得y_min最小的在前面
    x1, y1 = sorted(newBox[:2], key=lambda x: x[1])[0] 
    index = newBox.index([x1, y1]) # 取出(x_min,y_min)
    newBox.pop(index)
    
    newBox = sorted(newBox, key=lambda x: x[1],reverse=True)
    x4, y4 = sorted(newBox[:2], key=lambda x: x[0])[0]
    index = newBox.index([x4, y4])
    newBox.pop(index)
    
    newBox = sorted(newBox, key=lambda x: x[0],reverse=True)
    x2, y2 = sorted(newBox[:2], key=lambda x: x[1])[0]
    index = newBox.index([x2, y2])
    newBox.pop(index)

    newBox = sorted(newBox, key=lambda x: x[1],reverse=True)
    x3, y3 = sorted(newBox[:2], key=lambda x: x[0])[0]

    res = [[x1, y1], [x2, y2], [x3, y3], [x4, y4]]
    
    return res