数据分析--不同形式坐标点转换2

数据分析–不同形式坐标点转换2

在绘制非常多的矩形框的时候,先将坐标点全部转换成Polygon对象,提高代码运行效率,
如果要绘制成千上万张图片的时候,会极大地提高效率

下面提供两个坐标格式转换的方法

1. 将有序的8个点图像坐标转换成4面矩形坐标 用于创建Polygon对象的坐标形式
2. 将str格式点将两点坐标,左上角和右下角坐标转换成 4个点的顶点坐标 用于创建Polygon对象

复用性不强 仅做参考

def image_8points_to_box(image_points):
    """
    将有序的8个点图像坐标转换成4面矩形坐标 用于创建Polygon对象的坐标形式
    Args:
        image_points: 8个点坐标的列表 [p1,p2...p8] 对应位置顺序
                    upper_lt,upper_rt,upper_rb,upper_lb 为顶部顶点
                    lower_lt、lower_rt、lower_rb和lower_lb为底部顶点
    Returns:
        [Polygon1,Polygon2,Polygon3,Polygon4]
    """
    a0, a1, a2, a3, a4, a5, a6, a7 = image_points
    r1 = [a0, a1, a2, a3]
    r2 = [a4, a5, a6, a7]
    r3 = [a0, a4, a7, a3]
    r4 = [a1, a5, a6, a2]
    return [r1, r2, r3, r4]


def image_2points_to_box(image_points):
    """
    将str格式点将两点坐标,左上角和右下角坐标转换成 4个点的顶点坐标 用于创建Polygon对象
    Args:
        image_points: ["{'x1': a, 'x2': b, 'y1': c, 'y2': d}",...]

    Returns:
        [Polygon1,Polygon2,Polygon3,Polygon4]
    """

    def inner(p):
        try:
            if not isinstance(p, str):
                return
            p = eval(p)
            x1, x2, y1, y2 = p.values()
            return [[x1, y1], [x2, y1], [x2, y2], [x1, y2]]
        except Exception as e:
            print(e)
            print(p)

    ps = [inner(r) for r in image_points]
    return list(filter(None, ps))