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

相关内容

热门资讯

安卓系统能转什么系统好,探索最... 你有没有想过,你的安卓手机是不是也能换换口味,体验一下其他系统的魅力呢?没错,今天就来聊聊这个话题:...
龙之狂热安卓系统,释放龙族狂热 亲爱的手机控们,你是否曾为拥有一款独特的安卓系统而疯狂?今天,就让我带你走进一个充满奇幻色彩的龙之狂...
vivo手机安卓系统怎么升级系... 亲爱的手机控们,你是不是也和我一样,对手机的新功能充满期待呢?尤其是vivo手机的用户,是不是也在想...
鸿蒙2.0退回安卓系统,一场系... 你知道吗?最近科技圈里可是炸开了锅,因为华为的鸿蒙2.0操作系统竟然要退回安卓系统了!这可不是一个简...
安卓系统怎么复制卡,安卓系统卡... 你有没有遇到过这种情况:手机里的照片、视频或者重要文件,突然想备份到电脑上,却发现安卓系统的卡复制功...
app兼容低安卓系统,打造全民... 你有没有发现,现在手机APP更新换代的速度简直就像坐上了火箭!不过,你知道吗?有些APP可是特别贴心...
中间安卓系统叫什么,中间安卓系... 你有没有想过,安卓系统里竟然还有一个中间的版本?没错,就是那个让很多手机用户既熟悉又陌生的版本。今天...
安卓怎么用os系统,利用And... 你有没有想过,你的安卓手机其实可以变身成一个功能强大的操作系统呢?没错,就是那个我们平时在电脑上使用...
pe系统安卓能做么,探索安卓平... 亲爱的读者,你是否曾好奇过,那款在安卓设备上大受欢迎的PE系统,它究竟能做什么呢?今天,就让我带你一...
安卓 打印机系统,安卓打印机系... 你有没有想过,家里的安卓手机和打印机之间竟然能建立起如此紧密的联系?没错,就是那个安卓打印机系统!今...
安卓系统视频做铃声,轻松将视频... 你有没有想过,手机里那些动人心弦的视频,竟然可以变成手机铃声?没错,就是那种一响起就能瞬间抓住你耳朵...
海信电视安卓系统更新,畅享智能... 亲爱的电视迷们,你是否也像我一样,对家里的那台海信电视充满了期待?最近,海信电视安卓系统迎来了一次大...
安卓系统网页不能载入,排查与解... 最近是不是你也遇到了安卓系统网页不能载入的烦恼?别急,让我来帮你一探究竟,找出解决之道!一、问题现象...
赛欧3安卓系统,智能出行新体验 你有没有发现,现在的汽车越来越智能了?这不,我最近就发现了一款特别有意思的车型——赛欧3,它竟然搭载...
能装安卓系统吗,哪些设备能轻松... 你有没有想过,那些看起来普普通通的平板电脑,其实里面藏着大大的秘密呢?没错,就是能装安卓系统!今天,...
安卓能变苹果系统吗,技术揭秘与... 你有没有想过,安卓手机能不能变成苹果系统呢?这听起来就像是科幻电影里的情节,但今天,我们就来揭开这个...
车载安卓系统好卡,车载安卓系统... 你有没有遇到过这样的情况?你的车载安卓系统突然变得超级卡,就像蜗牛一样慢吞吞的,让人抓狂!没错,我就...
安卓系统怎样删除固件,轻松优化... 你有没有遇到过这种情况:手机里的安卓系统突然变得卡顿,或者某个固件版本让你觉得不爽,想要重新来过?别...
安卓鸿蒙系统视频对比,性能与体... 亲爱的读者们,你是否也像我一样,对安卓和鸿蒙系统之间的较量充满了好奇?今天,就让我们一起揭开这场科技...
电脑安卓系统横评,横跨性能与体... 你有没有想过,你的手机和电脑,其实就像两个超级英雄,各有各的本领和特点?今天,就让我带你来一场电脑安...