图片模糊处理批量生产模糊数据集

文件结构1如下图所示

# coding: utf-8
import numpy as np
import cv2 as cv
import os
 
def motion_blur(image, degree=30 ,angle=45):
    image = np.array(image)
 
    # 这里生成任意角度的运动模糊kernel的矩阵， degree越大，模糊程度越高
    M = cv.getRotationMatrix2D((degree / 2, degree / 2), angle, 1)
    motion_blur_kernel = np.diag(np.ones(degree))
    motion_blur_kernel = cv.warpAffine(motion_blur_kernel, M, (degree, degree))
 
    motion_blur_kernel = motion_blur_kernel / degree
    blurred = cv.filter2D(image, -1, motion_blur_kernel)
 
    # convert to uint8
    cv.normalize(blurred, blurred, 0, 255, cv.NORM_MINMAX)
    blurred = np.array(blurred, dtype=np.uint8)
    return blurred
#批量读取并处理
filePath = 'C:\\Users\\陈东祥\\Desktop\\新建文件夹 (4)\\abc\\'
for i in os.listdir(filePath):
    img = cv.imread("./abc/"+i)
    img_ = motion_blur(img)
    cv.imwrite(i,img_)