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

相关内容

热门资讯

省电手机推荐安卓系统,安卓系统... 手机这玩意儿,对于我们这些手机控来说,简直就是生活的必需品。但是,你知道吗?现在市面上那么多手机,要...
安卓系统衰落怎么恢复,探寻衰落... 你有没有发现,最近安卓系统好像有点儿“水土不服”了呢?曾经的霸主地位,如今似乎有些动摇。不过别急,今...
安卓系统手机应用锁,安全无忧的... 你有没有发现,现在手机里的秘密越来越多,是不是也跟小秘密一样,想要找个地方藏起来呢?没错,今天就要来...
安卓系统书院源app,安卓系统... 你有没有发现,手机里的安卓系统越来越智能了?今天,我要给你介绍一个特别有意思的书院源app,它可是安...
安卓系统8.1平板推荐,安卓8... 你有没有想过,拥有一款性能卓越、体验流畅的安卓系统8.1平板,简直就是移动办公和娱乐的完美搭档?没错...
谷歌不给华为安卓系统,探索替代... 你知道吗?最近科技圈可是炸开了锅!谷歌突然宣布,不给华为提供安卓系统了!这可不仅仅是两家公司之间的小...
选择安卓系统原因调查,揭秘安卓... 你有没有想过,为什么那么多人会选择安卓系统呢?是不是好奇他们到底看中了安卓的哪些“小秘密”?今天,就...
安卓系统的安全证书,守护移动安... 你知道吗?在咱们这个科技飞速发展的时代,手机已经成了我们生活中不可或缺的好伙伴。而说起手机,安卓系统...
谷歌安卓系统挣钱吗,如何通过它... 你有没有想过,那个无处不在的谷歌安卓系统,它到底是怎么赚钱的呢?没错,就是那个让我们的手机、平板、智...
GALGAME安卓换苹果系统,... 你有没有想过,那些在安卓手机上玩得如痴如醉的GALGAME,竟然也能在苹果系统上大放异彩?没错,就是...
华为宣布摆脱安卓系统,迈向自主... 哇,你知道吗?最近华为可是闹出了一个大新闻!那就是他们宣布要摆脱安卓系统,自己研发一套全新的操作系统...
安卓车机系统hcn,智能驾驶的... 你有没有发现,现在越来越多的汽车开始搭载智能化的车机系统了?这不,安卓车机系统HCN(Harmony...
超大平板安卓系统,创新体验与无... 你有没有想过,在这个科技飞速发展的时代,拥有一块超大平板电脑是多么酷炫的事情啊!想象那宽广的屏幕,无...
电脑装安卓系统步骤,电脑安装安... 你有没有想过,把安卓系统装在你的电脑上,是不是就像给电脑换了个新衣裳,瞬间变得时尚又实用呢?没错,今...
安卓系统视频存储软件,高效管理... 手机里的视频越来越多,是不是感觉存储空间不够用了?别急,今天就来给你安利几款安卓系统视频存储软件,让...
安卓系统如何连接奔驰,安卓系统... 你有没有想过,你的安卓手机和奔驰车之间也能来个亲密接触呢?没错,就是那种无缝连接的感觉,让你的手机和...
安卓系统emoji表情很少,探... 你知道吗?在手机世界里,emoji表情可是沟通的利器,它能让我们的聊天更加生动有趣。但是,你知道吗?...
安卓系统的德州游戏,刺激竞技体... 你有没有发现,安卓手机上的游戏世界越来越精彩了?尤其是那些德州游戏,简直让人停不下来!今天,就让我带...
xbox360安卓系统,探索跨... 你有没有想过,家里的老Xbox 360竟然也能变身成为安卓系统的超级玩家呢?没错,就是那个曾经陪伴我...
安卓系统怎么找云端,而是通过安... 你有没有想过,你的安卓手机里那些珍贵的照片、文件和笔记,其实都可以存放在云端,随时随地都能访问呢?没...