Python OpenCV 实现图像旋转且不裁剪图片的方法

旋转不裁剪的实现原理

要实现旋转且不裁剪图片，核心在于解决两个问题：一是计算旋转后图像所需的最小外接矩形尺寸，二是调整旋转矩阵以将图像中心重新对齐到新画布的中心。OpenCV 提供了 cv2.getRotationMatrix2D 函数来获取旋转矩阵，但该函数默认假设输出图像尺寸与原图一致。因此，我们需要先根据旋转角度计算新的宽度和高度，然后修改矩阵中的平移分量。

代码实现步骤

以下是完整的 Python 实现代码。该函数接收 OpenCV 读取的图像对象和旋转角度（单位为度）作为参数，返回旋转后的新图像。

import cv2
import numpy as np

def rotate_bound(image, angle):
    # 获取图像的高度和宽度
    (h, w) = image.shape[:2]
    # 计算图像的中心点
    (cX, cY) = (w // 2, h // 2)

    # 获取旋转矩阵
    # 注意：OpenCV 中角度为负值表示顺时针旋转
    M = cv2.getRotationMatrix2D((cX, cY), -angle, 1.0)

    # 计算旋转矩阵中的正弦和余弦值
    cos = np.abs(M[0, 0])
    sin = np.abs(M[0, 1])

    # 计算旋转后图像的新边界尺寸
    # 新宽度 = 原高度 * |sin| + 原宽度 * |cos|
    # 新高度 = 原高度 * |cos| + 原宽度 * |sin|
    nW = int((h * sin) + (w * cos))
    nH = int((h * cos) + (w * sin))

    # 调整旋转矩阵以考虑平移
    # 将旋转中心从原图中心移动到新画布中心
    M[0, 2] += (nW / 2) - cX
    M[1, 2] += (nH / 2) - cY

    # 执行仿射变换并返回结果图像
    return cv2.warpAffine(image, M, (nW, nH))

关键参数解析

在 cv2.getRotationMatrix2D 中，第三个参数 1.0 表示缩放因子为 1，即不进行缩放。旋转角度为 -angle，这是因为 OpenCV 中正值表示逆时针旋转，而负值表示顺时针旋转。通过取矩阵元素的绝对值来计算 cos 和 sin，可以确保无论旋转角度如何，计算出的新尺寸始终为正数。

应用场景

该方法特别适用于深度学习中的数据增强环节。在训练模型时，旋转图像可以增加数据的多样性，提高模型的泛化能力。通过保持图像不被裁剪，可以确保所有原始信息都被保留，避免因裁剪导致的特征丢失。

以上就是 Python OpenCV 实现图像旋转且不裁剪图片的详细内容，更多关于 OpenCV 图像处理的技术资料请关注本站其它相关文章！