利用opencv对训练数据图片进行旋转、翻折、放大、模糊、平移、增加噪点操作进行数据扩充

最新推荐文章于 2023-05-30 15:07:45 发布

komorebi_fresh

最新推荐文章于 2023-05-30 15:07:45 发布

阅读量761

点赞数 3

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

本文链接：https://blog.csdn.net/weixin_44270815/article/details/105340549

版权

该博客介绍了如何利用opencv库对深度学习训练数据集进行预处理，包括图像的旋转、翻转、放大、模糊、平移及添加噪声等操作，以实现数据扩充，提升模型的泛化能力。

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

博客原文链接：https://blog.csdn.net/weixin_44270815/article/details/105340549


```python
# -*- coding: utf-8 -*-
"""
Created on Sun Apr  5 23:12:45 2020

@author: xbb
"""
import cv2
import numpy as np
import random
import glob

dir_path='C:/Users/hp/Desktop/1/' #读取图片路径
dir_path2='C:/Users/hp/Desktop/out/' #保存图片路径
img=cv2.imread(dir_path)

num=1
out_hight=512
out_wight=512

def salt(src,percentage):
    NoiseImg=src
    rows,cols,_=NoiseImg.shape
    NoiseNum=int(percentage*rows*cols)
    for i in range(NoiseNum):
        randX=np.random.randint(0,rows-1)
        randY=np.random.randint(0,cols-1)
        if random.randint(0,1)<=0.5:
            NoiseImg[randX,randY]=255
        else:
            NoiseImg[randX,randY]=NoiseImg[randX,randY]

    return NoiseImg

for im in glob.glo