# -*- coding: utf-8 -*-
import sys
reload(sys) # Python2.5 初始化后会删除 sys.setdefaultencoding 这个方法，我们需要重新载入
sys.setdefaultencoding('utf-8') import os
import time
from PIL import Image,ImageChops,ImageEnhancedef image_reversal(img,savefilepath,save_filename):""" 图像翻转"""lr=img.transpose(Image.FLIP_LEFT_RIGHT) # 左右翻转ud=img.transpose(Image.FLIP_TOP_BOTTOM) # 上下翻转lr.save(savefilepath+save_filename)ud.save(savefilepath+save_filename)def image_rotation(img,savefilepath,save_filename):"""图像旋转"""out1=img.rotate(40) # 旋转20度out2=img.rotate(30) # 旋转30度out1.save(savefilepath+save_filename)out2.save(savefilepath+save_filename)def image_translation(img,savefilepath,save_filename):"""图像平移"""out3=ImageChops.offset(img,20,0) # 只沿X轴平移out4=ImageChops.offset(img,0,20) # 只沿y轴平移out3.save(savefilepath+save_filename)out4.save(savefilepath+save_filename)def image_brightness(img,savefilepath,save_filename):"""亮度调整"""bri=ImageEnhance.Brightness(img)bri_img1=bri.enhance(0.8) # 小于1为减弱bri_img2=bri.enhance(1.2) # 大于1为增强bri_img1.save(savefilepath+save_filename)bri_img2.save(savefilepath+save_filename)def image_chroma(img,savefilepath,save_filename):"""色度调整"""col = ImageEnhance.Color(img)col_img1 = col.enhance(0.7) # 色度减弱col_img2 = col.enhance(1.3) # 色度增强col_img1.save(savefilepath+save_filename)col_img2.save(savefilepath+save_filename)def image_contrast(img,savefilepath,save_filename):"""对比度调整"""con=ImageEnhance.Contrast(img)con_img1=con.enhance(0.7) # 对比度减弱con_img2=con.enhance(1.3) # 对比度增强con_img1.save(savefilepath+save_filename)con_img2.save(savefilepath+save_filename)def image_sharpness(img,savefilepath,save_filename):"""锐度调整"""sha = ImageEnhance.Sharpness(img)sha_img1 = sha.enhance(0.5) # 锐度减弱sha_img2 = sha.enhance(1.5) # 锐度增强sha_img1.save(savefilepath+save_filename)sha_img2.save(savefilepath+save_filename)# 定义扩充图片函数
def image_expansion(filepath,savefilepath,save_prefix):""":param filepath: 图片路径:param savefilepath: 扩充保存图片路径:param save_prefix: 图片前缀:return: 图片扩充数据集"""i = 1for parent, dirnames, filenames in os.walk(filepath):for filename in filenames:image_path=filepath+filenameprint('正在扩充图片：%s' %filename)try:img=Image.open(image_path)if img.mode == "P":img = img.convert('RGB')image_reversal(img,savefilepath,save_filename=save_prefix + str(i) + '.jpg')i += 1image_rotation(img,savefilepath,save_filename=save_prefix+str(i)+'.jpg')i += 1image_translation(img,savefilepath,save_filename=save_prefix+str(i)+'.jpg')i += 1image_brightness(img,savefilepath,save_filename=save_prefix+str(i)+'.jpg')i += 1image_chroma(img,savefilepath,save_filename=save_prefix+str(i)+'.jpg')i += 1image_contrast(img,savefilepath,save_filename=save_prefix+str(i)+'.jpg')i += 1image_sharpness(img,savefilepath,save_filename=save_prefix+str(i)+'.jpg')i += 1except Exception as e:print(e)passif __name__ == '__main__':# 设置图片路径filepath = '/data_1/lstm_general/data/数据增减记录/训练集/20200903-增加珠海重庆检验报告车牌数据/车牌车架号/'# 设置扩充保存图片路径savefilepath ='/data_1/lstm_general/data/数据增减记录/训练集/20200903-增加珠海重庆检验报告车牌数据/未命名文件夹 2/'# 设置前缀图片名称save_prefix='ticket_0_'time1 = time.time()image_expansion(filepath, savefilepath,save_prefix)time2 = time.time()print('总共耗时：' + str(time2 - time1) + 's')
本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！
