python旋转框裁剪代码

以下是 Python 中旋转框裁剪的示例代码：

import cv2
import numpy as np

def rotate_image(image, angle):
    """ 旋转图像 """
    height, width = image.shape[:2]
    rotation_matrix = cv2.getRotationMatrix2D((width/2, height/2), angle, 1)
    rotated_image = cv2.warpAffine(image, rotation_matrix, (width, height))
    return rotated_image

def rotate_box(box, angle, width, height):
    """ 旋转框 """
    x1, y1, x2, y2, x3, y3, x4, y4 = box
    # 将旋转中心点移动到图像中央
    cx, cy = width // 2, height // 2
    x1 -= cx
    y1 -= cy
    x2 -= cx
    y2 -= cy
    x3 -= cx
    y3 -= cy
    x4 -= cx
    y4 -= cy
    # 计算旋转后的坐标
    radians = angle * np.pi / 180
    cosine = np.cos(radians)
    sine = np.sin(radians)
    x1r = x1 * cosine - y1 * sine
    y1r = x1 * sine + y1 * cosine
    x2r = x2 * cosine - y2 * sine
    y2r = x2 * sine + y2 * cosine
    x3r = x3 * cosine - y3 * sine
    y3r = x3 * sine + y3 * cosine
    x4r = x4 * cosine - y4 * sine
    y4r = x4 * sine + y4 * cosine
    # 将旋转中心点移回原来位置
    x1r += cx
    y1r += cy
    x2r += cx
    y2r += cy
    x3r += cx
    y3r += cy
    x4r += cx
    y4r += cy
    return x1r, y1r, x2r, y2r, x3r, y3r, x4r, y4r

def clip_box(box, width, height):
    """ 裁剪框 """
    x1, y1, x2, y2, x3, y3, x4, y4 = box
    x1 = max(0, min(x1, width))
    y1 = max(0, min(y1, height))
    x2 = max(0, min(x2, width))
    y2 = max(0, min(y2, height))
    x3 = max(0, min(x3, width))
    y3 = max(0, min(y3, height))
    x4 = max(0, min(x4, width))
    y4 = max(0, min(y4, height))
    return x1, y1, x2, y2, x3, y3, x4, y4

def rotate_clip_box(box, angle, width, height):
    """ 旋转框裁剪 """
    rotated_box = rotate_box(box, angle, width, height)
    clipped_box = clip_box(rotated_box, width, height)
    return clipped_box

其中 `rotate_image` 函数用于旋转图像，`rotate_box` 函数用于旋转框，`clip_box` 函数用于裁剪框，`rotate_clip_box` 函数用于旋转框裁剪。这些函数可以根据实际情况进行调用和组合使用。