数据增强(扩充)适合初学者
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))

相关内容

热门资讯

神武3安卓跟苹果系统,安卓与苹... 你有没有发现,最近手机游戏界又掀起了一股热潮?没错,就是那款让人欲罢不能的神武3!不过,说到玩这款游...
安卓手机双系统好处,安卓手机的... 你有没有想过,你的安卓手机其实可以拥有双重身份呢?没错,就是那个听起来就让人兴奋的“双系统”!今天,...
咋录音软件推荐安卓系统,轻松捕... 你有没有想过,有时候想要记录下生活中的点点滴滴,或者制作一段属于自己的音乐作品,一款好用的录音软件可...
安卓7.1系统好不好,体验升级... 你有没有听说安卓7.1系统最近在手机圈里可是火得一塌糊涂呢?这不,我就来给你好好捋一捋,看看这个系统...
安卓系统到底为何卡顿,性能瓶颈... 你有没有发现,你的安卓手机有时候就像蜗牛一样慢吞吞的,让你等得心急火燎?别急,今天就来揭开安卓系统卡...
安卓系统如何自定图标,Andr... 你有没有发现,手机里的图标有时候看起来有点单调,甚至有点无聊呢?别急,今天就来教你怎么给安卓系统的图...
安卓系统的设置密码在哪,安卓系... 手机里的秘密太多啦,有时候连自己都忘了密码是啥!别急,今天就来手把手教你,安卓系统里的设置密码到底藏...
hk1box安卓原生系统和安卓... 亲爱的读者们,你是否曾为寻找一款既能满足日常使用,又能畅享大屏娱乐的安卓系统而烦恼?今天,就让我带你...
安卓手机系统崩溃蓝屏,揭秘原因... 手机突然蓝屏了,是不是瞬间感觉世界都安静了?别慌,今天就来和你聊聊安卓手机系统崩溃蓝屏的那些事儿。你...
平板电脑鸿蒙和安卓系统,平板电... 你有没有发现,最近平板电脑市场又热闹起来了呢?各大品牌纷纷推出新品,而其中最引人注目的莫过于鸿蒙系统...
安卓系统的帮助与反馈,提升用户... 你有没有发现,使用安卓系统的时候,有时候会遇到一些小麻烦,但别担心,安卓系统可是非常贴心的,它总有一...
安卓11系统4g,4G性能优化... 你知道吗?最近安卓系统又更新啦!这次可是安卓11系统哦,而且它还支持4G网络呢!是不是很激动?那就跟...
安卓系统怎么远程删除,远程删除... 你是不是也有过这样的经历:手机里的文件不小心误删了,或者某个应用突然变得异常,你想要远程清除它,但又...
安卓系统点歌软件app,打造个... 你有没有发现,现在手机里装满了各种APP,而其中有一个小家伙,几乎成了安卓手机用户的“音乐小助手”,...
p40安卓系统下载,下载与体验... 你有没有发现,最近你的手机是不是有点儿慢吞吞的?别急,别急,让我来给你支个招儿——升级你的P40安卓...
vivoiqoo3的安卓系统,... 你有没有发现,最近手机圈里又掀起了一股热潮?没错,就是vivo旗下的子品牌iQOO的新款手机——iQ...
系统可以直接还原吗安卓,轻松恢... 你有没有想过,手机里的那些照片、视频、联系人,如果突然消失了,那得多难过啊!别担心,今天就来聊聊这个...
安卓系统管控怎么解开,安卓系统... 你是不是也遇到了安卓系统管控的烦恼?手机被限制了,感觉自由受限,是不是超级郁闷?别急,今天就来手把手...
华为升级系统安卓11,开启智能... 你知道吗?最近华为的大动作可不少呢!他们家的新款手机升级了安卓11系统,这可真是让人眼前一亮啊!咱们...
安卓系统木马病毒生成,生成原理... 安卓系统木马病毒生成:揭秘网络威胁的新面孔在数字化时代,智能手机已经成为了我们生活中不可或缺的一部分...