数据增强，原始图像和xml文件旋转

番茄就要炒鸡蛋

于 2021-06-02 20:46:12 发布

阅读量1.4k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： opencv 文章标签： python 计算机视觉深度学习 xml opencv

本文链接：https://blog.youkuaiyun.com/hejunran/article/details/117482060

博客介绍了让image和xml按相同角度翻转的操作，可在目标识别中针对小样本进行数据增强。还提到若要更改保存路径，修改rotated_path即可，同时指出旋转后图像大小会改变，并展示了原始与翻转后的图像及xml。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使得image和xml按着相同的角度进行翻转，可以在目标识别中针对小样本进行一定的数据增强操作

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

最低0.47元/天解锁文章