脚本宝典收集整理的这篇文章主要介绍了样本扩增,脚本宝典觉得挺不错的,现在分享给大家,也给大家做个参考。
改变亮度、对比第、角度、镜像
from PIL import ImageEnhance import os import numpy as np from PIL import Image def brightnessEnhancement(root_path,img_name):#亮度增强 image = Image.open(os.path.join(root_path, img_name)) enh_bri = ImageEnhance.Brightness(image) # brightness = 1.1+0.4*np.random.random()#取值范围1.1-1.5 brightness = 1.5 image_brightened = enh_bri.enhance(brightness) return image_brightened def contrastEnhancement(root_path, img_name): # 对比度增强 image = Image.open(os.path.join(root_path, img_name)) enh_con = ImageEnhance.Contrast(image) # contrast = 1.1+0.4*np.random.random()#取值范围1.1-1.5 contrast = 1.5 image_contrasted = enh_con.enhance(contrast) return image_contrasted def rotation(root_path, img_name): img = Image.open(os.path.join(root_path, img_name)) random_angle = np.random.randint(-2, 2)*90 if random_angle==0: rotation_img = img.rotate(-90) #旋转角度 else: rotation_img = img.rotate( random_angle) # 旋转角度 # rotation_img.save(os.path.join(root_path,img_name.split('.')[0] + '_rotation.bmp')) return rotation_img def flip(root_path,img_name): #翻转图像 img = Image.open(os.path.join(root_path, img_name)) filp_img = img.transpose(Image.FLIP_LEFT_RIGHT) #左右翻转 # filp_img.save(os.path.join(root_path,img_name.split('.')[0] + '_flip.bmp')) return filp_img def createImage(imageDir,saveDir): i=0 for name in os.listdir(imageDir): i=i+1 saveName="cesun"+str(i)+".bmp" saveImage=contrastEnhancement(imageDir,name) saveImage.save(os.path.join(saveDir,saveName)) saveName1 = "flip" + str(i) + ".bmp" saveImage1 = flip(imageDir,name) saveImage1.save(os.path.join(saveDir, saveName1)) saveName2 = "brightnessE" + str(i) + ".bmp" saveImage2 = brightnessEnhancement(imageDir, name) saveImage2.save(os.path.join(saveDir, saveName2)) saveName3 = "rotate" + str(i) + ".bmp" saveImage = rotation(imageDir, name) saveImage.save(os.path.join(saveDir, saveName3)) if __name__ == '__main__': # 设置图片路径 filepath = 'C:/Users/Administrator/Desktop/images' # 设置扩充保存图片路径 savefilepath = 'C:/Users/Administrator/Desktop/imagesExp' createImage(filepath,savefilepath)
以上是脚本宝典为你收集整理的样本扩增全部内容,希望文章能够帮你解决样本扩增所遇到的问题。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。