数据增强(扩充)适合初学者
admin
2024-02-26 13:15:39
0

下面的代码可以实现13种数据增强,出自 PIL库的 Image和 ImageEnhance,后续如果想增加扩充倍数,可直接添加。

目录

1.ImageEnhance库:

2.Image库:

二、注意的问题:

三、代码展示:


一、相关的库主要包括:

1.ImageEnhance库:

ImageEnhance.enhance(factor) 对选择属性的数值增强factor倍 
ImageEnhance.Color(im) 调整图像的颜色平衡 
ImageEnhance.Contrast(im) 调整图像的对比度 
ImageEnhance.Brightness(im) 调整图像的亮度 
ImageEnhance.Sharpness(im) 调整图像的锐度 

2.Image库:

image.rotate()方法进行旋转

image.transpose()方法进行翻转

二、注意的问题:

将图片的原来文件夹和增强后的文件夹填写正确。

三、代码展示:


###
#本代码共采用了13种数据增强,如采用其他数据增强方式,可以参考本代码,随意替换。
#imageDir 为原数据集的存放位置
#saveDir  为数据增强后数据的存放位置
###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.jpg'))return filp_imgdef rotation(root_path, img_name):img = Image.open(os.path.join(root_path, img_name))rotation_img = img.rotate(20) #旋转角度20# rotation_img.save(os.path.join(root_path,img_name.split('.')[0] + '_rotation.jpg'))return rotation_imgdef rotation2(root_path, img_name):img = Image.open(os.path.join(root_path, img_name))rotation_img = img.rotate(10) #旋转角度10# rotation_img.save(os.path.join(root_path,img_name.split('.')[0] + '_rotation.jpg'))return rotation_imgdef rotation3(root_path, img_name):img = Image.open(os.path.join(root_path, img_name))rotation_img = img.rotate(90) #旋转角度90# rotation_img.save(os.path.join(root_path,img_name.split('.')[0] + '_rotation.jpg'))return rotation_imgdef rotation4(root_path, img_name):img = Image.open(os.path.join(root_path, img_name))rotation_img = img.rotate(180) #旋转角度180# rotation_img.save(os.path.join(root_path,img_name.split('.')[0] + '_rotation.jpg'))return rotation_imgdef rotation5(root_path, img_name):img = Image.open(os.path.join(root_path, img_name))rotation_img = img.rotate(45) #旋转角度45# rotation_img.save(os.path.join(root_path,img_name.split('.')[0] + '_rotation.jpg'))return rotation_imgdef randomColor(root_path, img_name): #随机颜色"""对图像进行颜色抖动:param image: PIL的图像image:return: 有颜色色差的图像image"""image = Image.open(os.path.join(root_path, img_name))random_factor = np.random.randint(0, 31) / 10.  # 随机因子color_image = ImageEnhance.Color(image).enhance(random_factor)  # 调整图像的饱和度random_factor = np.random.randint(10, 21) / 10.  # 随机因子brightness_image = ImageEnhance.Brightness(color_image).enhance(random_factor)  # 调整图像的亮度random_factor = np.random.randint(10, 21) / 10.  # 随机因子contrast_image = ImageEnhance.Contrast(brightness_image).enhance(random_factor)  # 调整图像对比度random_factor = np.random.randint(0, 31) / 10.  # 随机因子return ImageEnhance.Sharpness(contrast_image).enhance(random_factor)  # 调整图像锐度def contrastEnhancement(root_path, img_name):  # 对比度增强image = Image.open(os.path.join(root_path, img_name))enh_con = ImageEnhance.Contrast(image)contrast = 1.5image_contrasted = enh_con.enhance(contrast)return image_contrasteddef contrastEnhancement2(root_path, img_name):  # 对比度增强2image = Image.open(os.path.join(root_path, img_name))enh_con = ImageEnhance.Contrast(image)contrast = 1.75image_contrasted = enh_con.enhance(contrast)return image_contrasteddef brightnessEnhancement(root_path,img_name):#亮度增强image = Image.open(os.path.join(root_path, img_name))enh_bri = ImageEnhance.Brightness(image)brightness = 1.5image_brightened = enh_bri.enhance(brightness)return image_brighteneddef colorEnhancement(root_path,img_name):#颜色增强image = Image.open(os.path.join(root_path, img_name))enh_col = ImageEnhance.Color(image)color = 1.5image_colored = enh_col.enhance(color)return image_coloreddef colorEnhancement2(root_path,img_name):#颜色增强->黑白图image = Image.open(os.path.join(root_path, img_name))enh_col = ImageEnhance.Color(image)color = 2image_colored = enh_col.enhance(color)return image_colored#锐度增强
def colorSharpness(root_path,img_name):image = Image.open(os.path.join(root_path, img_name))enh_col = ImageEnhance.Sharpness(image)sharpness = 3.0image_colored = enh_col.enhance(sharpness)return image_coloredfrom PIL import Image
from PIL import ImageEnhance
import os
import cv2
import numpy as np#D:\images_background\22 (2)
imageDir="D:/学习/人工智能/孪生网络/数据集/原数据集 - 轮胎痕迹/trace.6L"     #要改变的图片的路径文件夹
saveDir="D:/学习/人工智能/孪生网络/数据集/原数据集 - 轮胎痕迹/trace.6L"   #要保存的图片的路径文件夹
#D:\trace_images01\.6LD:\学习\人工智能\孪生网络\数据集\原数据集 - 轮胎花纹\t (2)
for name in os.listdir(imageDir):saveName= name[:-4]+"bright.jpg"saveImage=brightnessEnhancement(imageDir,name)#2.亮度增强saveImage.save(os.path.join(saveDir,saveName))saveName= name[:-4]+"flip.jpg"saveImage=flip(imageDir,name)#3.翻转图像saveImage.save(os.path.join(saveDir,saveName))saveName= name[:-4]+"rotation.jpg"saveImage=rotation(imageDir,name)#4.旋转角度20度saveImage.save(os.path.join(saveDir,saveName))saveName= name[:-4]+"randmcolor.jpg"saveImage=randomColor(imageDir,name)#5.随机颜色saveImage.save(os.path.join(saveDir,saveName))saveName= name[:-4]+"color.jpg"saveImage=colorEnhancement(imageDir,name)#6.颜色增强saveImage.save(os.path.join(saveDir,saveName))saveName= name[:-4]+"contrast.jpg"saveImage=contrastEnhancement(imageDir,name) #7.对比度增强saveImage.save(os.path.join(saveDir,saveName))saveName= name[:-4]+"rotation3.jpg"saveImage=rotation3(imageDir,name)#8.旋转角度30度saveImage.save(os.path.join(saveDir,saveName))saveName= name[:-4]+"rotation4.jpg"saveImage=rotation4(imageDir,name)#9.旋转角度45度saveImage.save(os.path.join(saveDir,saveName))saveName= name[:-4]+"contrast2.jpg"saveImage=contrastEnhancement2(imageDir,name) #10.对比度增强saveImage.save(os.path.join(saveDir,saveName))saveName= name[:-4]+"colorEnhancement2.jpg"saveImage=colorEnhancement2(imageDir,name) #11.黑白图saveImage.save(os.path.join(saveDir,saveName))saveName= name[:-4]+"colorSharpness.jpg"saveImage=colorSharpness(imageDir,name) #12.增强因子为2.0表示锐化图像。值越大图像边界越多越清晰。saveImage.save(os.path.join(saveDir,saveName))saveName= name[:-4]+"rotate5.jpg"saveImage=rotation5(imageDir,name) #13.对比度增强saveImage.save(os.path.join(saveDir,saveName))

相关内容

热门资讯

安卓系统的地图怎样下载,下载与... 你有没有发现,现在不管去哪里,手机地图都成了我们的好帮手?尤其是安卓系统的地图,功能强大,用起来超级...
安卓9.0系统挂机游戏,轻松享... 你有没有发现,自从安卓9.0系统更新后,手机里的游戏体验简直就像坐上了火箭!今天,就让我带你一起探索...
安卓系统怎么用迅雷下载,安卓系... 你有没有想过,在安卓系统上下载文件竟然也能这么简单?没错,今天就要来给你揭秘,如何用迅雷在安卓系统上...
安卓手机刷成学生系统,探索全新... 你有没有想过,你的安卓手机其实可以变身成一个充满学习氛围的学生系统呢?没错,就是那种看起来简洁、功能...
ios能迁移安卓系统吗,iOS... 你有没有想过,你的iPhone里的那些宝贝应用,能不能搬到安卓手机上继续使用呢?这可是不少手机用户的...
荣耀10安卓11系统,畅享极致... 你知道吗?最近手机界可是热闹非凡呢!荣耀10这款手机,自从升级到了安卓11系统,简直就像脱胎换骨了一...
安卓系统pc版电脑配置,打造流... 你有没有想过,安卓系统竟然也能在电脑上运行呢?没错,就是那个我们手机上常用的安卓系统,现在也能在PC...
tcllinux系统刷安卓系统... 你有没有想过,你的TCL Linux系统竟然也能升级成安卓系统呢?没错,就是那个我们日常使用的安卓系...
安卓13系统更新蓝牙,蓝牙功能... 你有没有发现,最近你的安卓手机好像变得不一样了?没错,就是那个神秘的安卓13系统更新,它悄悄地来到了...
安卓系统钉钉打开声音,安卓系统... 你有没有遇到过这种情况?手机里装了钉钉,可每次打开它,那声音就“嗖”地一下跳出来,吓你一跳。别急,今...
理想汽车操作系统安卓,基于安卓... 你有没有想过,一辆汽车,除了能带你去你想去的地方,还能像智能手机一样,给你带来智能化的体验呢?没错,...
安卓系统越狱还能升级吗,升级之... 你有没有想过,你的安卓手机越狱后,还能不能愉快地升级系统呢?这可是不少手机爱好者关心的大问题。今天,...
安卓系统蓝牙耳机拼多多,畅享无... 你有没有发现,最近蓝牙耳机在市场上可是火得一塌糊涂呢!尤其是安卓系统的用户,对于蓝牙耳机的要求那可是...
安卓变苹果系统桌面,桌面系统变... 你知道吗?最近有个大新闻在科技圈里炸开了锅,那就是安卓用户纷纷转向苹果系统桌面。这可不是闹着玩的,这...
鸿蒙系统怎么下安卓,鸿蒙系统下... 你有没有想过,你的手机里那个神秘的鸿蒙系统,竟然也能和安卓世界来一场亲密接触呢?没错,今天就要来揭秘...
手机安卓系统流程排行,便捷操作... 你有没有发现,现在手机的世界里,安卓系统就像是个大舞台,各种版本层出不穷,让人眼花缭乱。今天,就让我...
安卓系统左上角hd,左上角HD... 你有没有发现,每次打开安卓手机,左上角那个小小的HD标识总是默默地在那里,仿佛在诉说着什么?今天,就...
安卓系统软件文件,架构解析与功... 你有没有发现,手机里的安卓系统软件文件就像是一个神秘的宝库,里面藏着无数的宝藏?今天,就让我带你一起...
安卓系统输入法回车,探索安卓输... 你有没有发现,在使用安卓手机的时候,输入法回车键的奇妙之处?它就像是我们指尖的魔法师,轻轻一点,文字...
安卓修改系统时间的软件,轻松掌... 你有没有想过,有时候手机上的时间不对劲,是不是觉得生活节奏都被打乱了?别急,今天就来给你揭秘那些神奇...