CNN Data Augmentation（数据增强）-旋转

最新推荐文章于 2024-10-22 22:39:43 发布

木_凌

最新推荐文章于 2024-10-22 22:39:43 发布

阅读量1.5w

点赞数 7

CC 4.0 BY-SA版权

分类专栏：深度学习小知识文章标签：数据增强 CNN 卷积神经网络深度学习

本文链接：https://blog.youkuaiyun.com/u014540717/article/details/53301195

1、原始状态

最初的图像是这个样子的
这里写图片描述
.xml文件张下面这个样子

<annotation>
    <object>
        <name>face</name>
        <difficult>0</difficult>
        <bndbox>
            <xmin>315.307918</xmin>
            <ymin>240.234604</ymin>
            <xmax>693.677419</xmax>
            <ymax>699.683284</ymax>
        </bndbox>
    </object>
</annotation>

2、旋转图像并修改对应的xml文件

import cv2
import math
import numpy as np
import os
# pdb仅仅用于调试，不用管它
import pdb

#旋转图像的函数
def rotate_image(src, angle, scale=1.):
    w = src.shape[1]
    h = src.shape[0]
    # 角度变弧度
    rangle = np.deg2rad(angle)  # angle in radians
    # now calculate new image width and height
    nw = (abs(np.sin(rangle)*h) + abs(n