Numpy：矩形框四个顶点顺时针排序

问题说明

一个矩形有四个顶点，设计一个函数将其进行顺时针排序，并返回。

代码

主要的思路就是先对矩形的四个点，按x方向进行一次排序后分为两组顶点，前两个为一组Left，后两个为一组Right。然后对Left进行排序，确保y值大的在前面；对RIght进行排序，确保y值小的在前面。拼接Left与Right即满足要求。

返回顺序左下、左上、右上、右下

def order_points(pts):
    ''' sort rectangle points by clockwise '''
    sort_x = pts[np.argsort(pts[:, 0]), :]
    
    Left = sort_x[:2, :]
    Right = sort_x[2:, :]
    # Left sort
    Left = Left[np.argsort(Left[:,1])[::-1], :]
    # Right sort
    Right = Right[np.argsort(Right[:,1]), :]
    
    return np.concatenate((Left, Right), axis=0)

测试

img = np.zeros((512, 512, 3), dtype=np.uint8)
pts = np.array([[128,128],[256,256],[256,100],[128,256]], dtype=np.int32)
pts = order_points(pts)
print(pts)

结果

[[128 256]
 [128 128]
 [256 100]
 [256 256]]

另一思路

如果非矩形，但是为凸四边形的话，可以使用这个思路进行设计。选择[x,y]值最小的点（左上角的点），将其与剩余三个点连线，连线后。如果存在一个点在线上方，一个点在线下方。那么线上的点为右上，线下的点为左下，连接点为右下。即可得到正确顺序。