图像增强（2）-旋转一定的角度，同时处理图片和标注好的xml文件

最新推荐文章于 2024-09-02 23:01:53 发布

浪迹天涯@wxy

最新推荐文章于 2024-09-02 23:01:53 发布

阅读量440

点赞数

分类专栏：脚本之家文章标签： xml python 计算机视觉

本文链接：https://blog.csdn.net/wxy2020915/article/details/126786473

版权

本文介绍如何在Python环境下使用Opencv进行图像旋转，并处理对应的XML标注文件。内容包括旋转前后的图像对比，以及操作过程中所需安装的软件环境和文件组织方式。

摘要由CSDN通过智能技术生成

代码：

import os
import numpy as np
import cv2
import math
import xml.etree.ElementTree as ET
from PIL import Image


# 按角度翻转图片
def rotate_img(src, angle, scale=1):
    width = src.shape[1]  # 原始图像的宽
    height = src.shape[0]  # 原始图像的高
    # 角度变弧度
    re_angle = np.deg2rad(angle)
    # 计算新图片的高度和宽度
    new_width = (abs(np.sin(re_angle) * height) + abs(np.cos(re_angle) * width)) * scale
    new_height = (abs(np.cos(re_angle) * height) + abs(np.sin(re_angle) * width)) * scale

    rotate_matrix = cv2.getRotationMatrix2D((new_width * 0.5, new_height * 0.5), angle, scale)
    rotate_move = np.dot(rotate_matrix, np.array([(new_width - width) * 0.5, (new_height - height) * 0.5, 0]))

    # update translation
    rotate_matrix[0, 2] += rotate_move[0]
    rotate_matrix[1, 2] += rotate_move[1]

    dst = cv2.warpAffine(img, rotate_matrix, (int(math.ceil(new_width)), int(math.ceil(new_height))),
                         flags=cv2.INTER_LANCZOS4)
    return dst


# 翻转后的xml文件信息
def rotate_xml(src, xmin, ymin, xmax, ymax, angle, scale=1):
    width = src.shape[1]
    height = src.shape[0]
    re_a