数据增强旋转

夏日清风有你

已于 2022-06-09 21:45:51 修改

阅读量409

点赞数

分类专栏： PyTorch numpy 文章标签： python 开发语言

于 2022-06-09 21:41:06 首次发布

本文链接：https://blog.csdn.net/fanlily913/article/details/125211620

版权

PyTorch 同时被 2 个专栏收录

50 篇文章 1 订阅

订阅专栏

numpy

23 篇文章 1 订阅

订阅专栏

本文介绍了如何使用Python的random和numpy模块生成随机数，并展示了如何运用这些随机数来实现图像数据增强，通过旋转操作增强图像数据的多样性。代码示例中展示了针对不同概率执行90度、180度和270度旋转的数据增强策略，这对于训练深度学习模型时扩大训练集的覆盖范围具有重要意义。

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

random 生成随机数

import random
import numpy as np
seed = 666
np.random.seed(seed)  # Numpy module.
random.seed(seed)  # Python random module.
for i in range(10):
   P = random.random()
   print(P)

数据增强旋转

def rotate_nobound(image, angle, center=None, scale=1.):
    (h, w) = image.shape[:2]
    # if the center is None, initialize it as the center of the image
    if center is None:
        center = (w // 2, h // 2)    # perform the rotation
    M = cv2.getRotationMatrix2D(center, angle, scale)
    rotated = cv2.warpAffine(image, M, (w, h))
    return rotated

数据增强应用

   p = random.random()
   if p <= 0.25:
        imgo_arr = rotate_nobound(imgo_arr, 90, center=None, scale=1.)  # 转为numpy 进行数据增强
        mask = rotate_nobound(mask, 90, center=None, scale=1.)  # 转为numpy 进行数据增强
    elif 0.25 < p and p<= 0.5:
        imgo_arr = rotate_nobound(imgo_arr, 180, center=None, scale=1.)  # 转为numpy 进行数据增强
        mask = rotate_nobound(mask, 180, center=None, scale=1.)  # 转为numpy 进行数据增强
    else: # 0.5 < p and p<= 0.75:
        imgo_arr = rotate_nobound(imgo_arr, 270, center=None, scale=1.)  # 转为numpy 进行数据增强
        mask = rotate_nobound(mask, 270, center=None, scale=1.)  # 转为numpy 进行数据增强