写一个随机旋转角度YOLO标签转换的程序

好的，以下是一个简单的随机旋转角度YOLO标签转换程序的示例代码：

import cv2
import numpy as np
import random

# 定义旋转函数
def rotate(image, angle):
    h, w = image.shape[:2]
    center = (w / 2, h / 2)
    M = cv2.getRotationMatrix2D(center, angle, 1.0)
    rotated = cv2.warpAffine(image, M, (w, h))
    return rotated, M

# 定义YOLO标签转换函数
def convert_yolo_label(label, angle):
    # 解析标签
    class_id, x, y, w, h = label.split()

    # 计算旋转后的坐标
    x, y = float(x), float(y)
    w, h = float(w), float(h)
    x1, y1 = x - w / 2, y - h / 2
    x2, y2 = x + w / 2, y - h / 2
    x3, y3 = x + w / 2, y + h / 2
    x4, y4 = x - w / 2, y + h / 2
    points = np.array([[x1, y1], [x2, y2], [x3, y3], [x4, y4]])
    rot_mat = cv2.getRotationMatrix2D((x, y), angle, 1)
    points = np.dot(rot_mat, np.vstack((points.T, np.ones(4))))
    x1, y1 = points[0][0], points[1][0]
    x2, y2 = points[0][1], points[1][1]
    x3, y3 = points[0][2], points[1][2]
    x4, y4 = points[0][3], points[1][3]
    x, y = (x1 + x3) / 2, (y1 + y3) / 2
    w, h = np.abs(x1 - x3), np.abs(y1 - y3)

    # 返回旋转后的标签
    return f"{class_id} {x} {y} {w} {h}"

# 加载图像和标签
image = cv2.imread("image.jpg")
labels = ["0 0.5 0.5 0.3 0.4", "1 0.2 0.7 0.4 0.2"]

# 随机生成旋转角度
angle = random.randint(0, 360)

# 旋转图像和标签
rotated_image, M = rotate(image, angle)
rotated_labels = [convert_yolo_label(label, angle) for label in labels]

# 显示旋转后的图像和标签
cv2.imshow("Rotated Image", rotated_image)
print("Rotated Labels:")
for label in rotated_labels:
    print(label)

cv2.waitKey(0)
cv2.destroyAllWindows()

该程序包含两个函数：rotate()函数用于旋转图像，convert_yolo_label()函数用于转换YOLO标签。

在程序中，我们首先加载图像和标签。然后，我们使用随机生成的旋转角度旋转图像，并使用旋转矩阵将YOLO标签转换为旋转后的坐标。最后，我们显示旋转后的图像和标签。

请注意，此代码仅供参考。在实际使用中，您可能需要根据您的数据集格式和需求进行修改。