目标检测标签格式转换（kitti数据集格式转四坐标）

最新推荐文章于 2025-04-14 21:24:03 发布

点PY

最新推荐文章于 2025-04-14 21:24:03 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏：目标检测文章标签： 1024程序员节

本文链接：https://blog.csdn.net/weixin_42990464/article/details/109253923

目标检测专栏收录该内容

37 篇文章

订阅专栏

这篇博客介绍了如何利用Python实现图像上的点旋转以及创建旋转形状的坐标。`efrotatePoint`函数用于计算点在给定角度下的旋转位置，而`addRotatedShape`函数则用于生成矩形四个顶点在旋转后的坐标。提供的样例数据展示了旋转前后的坐标变化，并提到了一个可视化的旋转框标注工具LabelImg-KITTI。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

ef rotatePoint(xc, yc, xp, yp, theta):
    xoff = xp - xc

    yoff = yp - yc

    cosTheta = math.cos(theta)

    sinTheta = math.sin(theta)

    pResx = cosTheta * xoff + sinTheta * yoff

    pResy = - sinTheta * xoff + cosTheta * yoff

    return xc + pResx, yc + pResy


def addRotatedShape(cx, cy, w, h, angle):
    p0x, p0y = rotatePoint(cx, cy, cx - w / 2, cy - h / 2, -angle)

    p1x, p1y = rotatePoint(cx, cy, cx + w / 2, cy - h / 2, -angle)

    p2x, p2y = rotatePoint(cx, cy, cx + w / 2, cy + h / 2, -angle)

    p3x, p3y = rotatePoint(cx, cy, cx - w / 2, cy + h / 2, -angle)

    return p0x, p0y, p1x, p1y, p2x, p2y, p3x, p3y

样例数据：
转换前：

2 284.755840 55.945635 17.589079 125.675577 -5.620411

转换后：

316.4876871422387, 1.0000121940895426, 330.3529509009749, 11.82268104095521, 253.02399285776124, 110.89125780591047, 239.15872909902504, 100.06858895904477

可视化：
在这里插入图片描述
旋转框标注工具：
https://github.com/zexihan/labelImg-kitti