根据坐标进行排序

小女孩真可爱

于 2024-05-27 21:38:22 发布

阅读量209

点赞数 1

分类专栏：便捷小代码文章标签： python 排序算法

本文链接：https://blog.csdn.net/m0_48095841/article/details/139248559

版权

便捷小代码专栏收录该内容

14 篇文章 0 订阅

订阅专栏

分享一个简单的坐标排序，可用于yolo输出的boxes进行排序。此排序适用于倾斜角度不大的情况，如果文本倾斜过多，排序结果可能有问题。

    def sort_point(point_list):
        # 输入的point_list是一个列表，格式[(x1,y1),(x2,y2),(x3,y3)]
        for i in range(len(point_list)):
            for j in range(i+1, len(point_list)):
                # 先比较y坐标，如果绝对值小于等于10，则认为这两个字符在同一行，然后比较x坐标
                if abs(point_list[i][1]-point_list[j][1]) < 10:
                    if point_list[i][0] > point_list[j][0]:
                        point_list[i], point_list[j] = point_list[j], point_list[i]
                # 如果y坐标差值大于10，则认为这两个字符不在同一行，直接比较y坐标
                else:
                    if point_list[i][1] > point_list[j][1]:
                        point_list[i], point_list[j] = point_list[j], point_list[i]
        return point_list