Python无第三方包用变换矩阵实现彩色图像0/90/180/270度旋转问题求助

• 浅拷贝问题：output_image = image 是引用赋值，修改输出图像会直接覆盖原图像数据，没有开辟新的存储空间。
• 坐标计算错误：循环内的x、y被写死为图像宽高的最大值，完全没有使用循环变量i、j，所有像素都基于同一个固定坐标计算变换位置，自然无法正确赋值。
• 坐标原点不匹配：标准旋转矩阵基于笛卡尔坐标系原点（中心）计算，但图像的坐标原点在左上角，未做坐标平移会导致变换后的坐标大量溢出有效范围。
• 输出尺寸不匹配：90°、270°旋转后图像的宽高会互换，你没有调整输出图像的尺寸，会出现坐标越界丢失像素的问题。

如果你要沿用变换矩阵的思路，修复后的可运行代码如下：

import math
import numpy as np

def rotate_image(image, rotate_angle):
    H, W = image.shape[:2]
    angle = math.radians(rotate_angle)
    # 计算旋转后输出图像的尺寸
    cos_val = abs(math.cos(angle))
    sin_val = abs(math.sin(angle))
    new_W = int(W * cos_val + H * sin_val)
    new_H = int(W * sin_val + H * cos_val)
    # 初始化新图像，避免覆盖原图像
    output_image = np.zeros((new_H, new_W, 3), dtype=image.dtype)
    # 原图像中心坐标
    center_x = W / 2
    center_y = H / 2
    # 新图像中心坐标
    new_center_x = new_W / 2
    new_center_y = new_H / 2

    for i in range(H):
        for j in range(W):
            # 平移到中心坐标系
            x = j - center_x
            y = i - center_y
            # 应用旋转矩阵
            x_n = x * math.cos(angle) + y * math.sin(angle)
            y_n = -x * math.sin(angle) + y * math.cos(angle)
            # 平移回图像左上角坐标系
            x_n = int(x_n + new_center_x)
            y_n = int(y_n + new_center_y)
            # 边界判断避免越界
            if 0 <= x_n < new_W and 0 <= y_n < new_H:
                output_image[y_n, x_n, :] = image[i, j, :]
    return output_image

因为你仅需要支持0°、90°、180°、270°四个固定角度，完全不需要用循环计算每个像素的变换，直接用numpy数组的转置、翻转操作即可，性能比双层循环提升上百倍：

import numpy as np

def rotate_image_fast(image, rotate_angle):
    # 仅支持0/90/180/270四个角度，取模后统一处理
    angle = rotate_angle % 360
    if angle == 0:
        return image.copy()
    elif angle == 90:
        # 顺时针90度：转置后上下翻转
        return image.transpose(1, 0, 2)[::-1, :, :].copy()
    elif angle == 180:
        # 180度：上下翻转后左右翻转
        return image[::-1, ::-1, :].copy()
    elif angle == 270:
        # 顺时针270度：转置后左右翻转
        return image.transpose(1, 0, 2)[:, ::-1, :].copy()
    else:
        raise ValueError("仅支持0/90/180/270度旋转")

该方案没有循环开销，完全基于numpy的底层C实现操作，运行效率极高，且不需要做复杂的坐标计算和边界判断。

内容的提问来源于stack exchange，提问作者Dilohn Lasdnaym