这次在做图像识别的项目中,由于部分选项的图片较为稀缺,所以采用了一些方法来扩容图片,图片加噪便是其中之一
当然也可以使用旋转,左右翻转的方式来进行扩容(上下翻转好像不行,未验证)
这些都是不错的手段来对图片数据进行扩容
import cv2
import os
import numpy as np
import glob
# 文件夹自己建一下
# 这个函数对于文件名字有一定的要求,懒得改了,如果报错了没有shape属性的错,可以试试换个文件图片名字
# 如果后续有大佬可以试试
# 定义参数
loadPath = './img/'
savePath = './saltImage/'
proname = "Rose" # 图片前置名
rate = 0.004
# 加噪函数
def saltpepper(img, n):
m = int((img.shape[0] * img.shape[1]) * n)
for a