元学习论文总结||小样本学习论文总结

2017-2019年计算机视觉顶会文章收录 AAAI2017-2019 CVPR2017-2019 ECCV2018 ICCV2017-2019 ICLR2017-2019 NIPS2017-2019

一：日志依赖

https://blog.csdn.net/weixin_41803874/article/details/81201807

数据操作完整源码自 vieo.zhu@foxmail.com 申请

二：代码实现

from diagnose_logging import Logger
from PIL import Image
import os# 声明日志
log = Logger('img_pre.py')
logger = log.getlog()class ImgPre:def __init__(self, rootPath, export_path_base):self.rootPath = rootPath  # 图像完整路径self.export_path_base = export_path_base# 创建输出根目录try:if not os.path.exists(export_path_base):os.mkdir(export_path_base)except Exception as e:logger.error(e)logger.info('ImgPre: %s', rootPath)def get_savename(self, operate):""":param export_path_base: 图像输出路径:param operate: 脸部区域名:return: 返回图像存储名"""try:import time# 获取时间戳，用于区分图像now = time.time()tail_time = str(round(now * 1000000))[-4:]  # 时间戳尾数head_time = time.strftime("%Y%m%d%H%M%S", time.localtime(time.time()))# 时间标签label = str(head_time + tail_time)# 输出文件夹export_path_base = self.export_path_base# 子文件夹以“操作operate”命名out_path = export_path_base + operate# 创建子文件夹if not os.path.exists(out_path):os.mkdir(out_path)# 存储完整路径savename = out_path + '/' + operate + '_' + label + ".jpg"# 日志logger.info('save:%s', savename)return savenameexcept Exception as e:logger.error('get_savename ERROR')logger.error(e)def lightness(self, light):"""改变图像亮度.推荐值：0.87，1.07明亮程度darker < 1.0 <lighter"""try:operate = 'lightness_' + str(light)# 图像完整路径rootPath = self.rootPathwith Image.open(rootPath) as image:# 图像左右翻转out = image.point(lambda p: p * light)# 重命名savename = self.get_savename(operate)# 图像存储out.save(savename)# 日志# logger.info(operate)except Exception as e:logger.error('ERROR %s', operate)logger.error(e)def rotate(self, angle):"""图像旋转15度、30度."""try:operate = 'rotate_' + str(angle)# 图像完整路径rootPath = self.rootPathwith Image.open(rootPath) as image:# 图像左右翻转out = image.rotate(angle)# 重命名savename = self.get_savename(operate)# 图像存储out.save(savename, quality=100)# 日志# logger.info(operate)except Exception as e:logger.error('ERROR %s', operate)logger.error(e)def transpose(self):"""图像左右翻转操作."""try:operate = 'transpose'# 图像完整路径rootPath = self.rootPathwith Image.open(rootPath) as image:# 图像左右翻转out = image.transpose(Image.FLIP_LEFT_RIGHT)# 重命名savename = self.get_savename(operate)# 图像存储out.save(savename, quality=100)  # quality=100# 日志# logger.info(operate)except Exception as e:logger.error('ERROR %s', operate)logger.error(e)def deform(self):"""图像拉伸."""try:operate = 'deform'# 图像完整路径rootPath = self.rootPathwith Image.open(rootPath) as image:w, h = image.sizew = int(w)h = int(h)# 拉伸成宽为w的正方形out_ww = image.resize((int(w), int(w)))savename = self.get_savename(operate + '_ww')out_ww.save(savename, quality=100)# 拉伸成宽为h的正方形out_ww = image.resize((int(h), int(h)))savename = self.get_savename(operate + '_hh')out_ww.save(savename, quality=100)# 日志# logger.info(operate)except Exception as e:logger.error('ERROR %s', operate)logger.error(e)def crop(self):"""提取四个角落和中心区域."""try:operate = 'crop'# 图像完整路径rootPath = self.rootPathwith Image.open(rootPath) as image:w, h = image.size# 切割后尺寸scale = 0.875# 切割后长宽ww = int(w * scale)hh = int(h * scale)# 图像起点，左上角坐标x = y = 0# 切割左上角x_lu = xy_lu = yout_lu = image.crop((x_lu, y_lu, ww, hh))savename = self.get_savename(operate + '_lu')out_lu.save(savename, quality=100)# logger.info(operate + '_lu')# 切割左下角x_ld = int(x)y_ld = int(y + (h - hh))out_ld = image.crop((x_ld, y_ld, ww, hh))savename = self.get_savename(operate + '_ld')out_ld.save(savename, quality=100)# logger.info(operate + '_ld')# 切割右上角x_ru = int(x + (w - ww))y_ru = int(y)out_ru = image.crop((x_ru, y_ru, w, hh))savename = self.get_savename(operate + '_ru')out_ru.save(savename, quality=100)# logger.info(operate + '_ru')# 切割右下角x_rd = int(x + (w - ww))y_rd = int(y + (h - hh))out_rd = image.crop((x_rd, y_rd, w, h))savename = self.get_savename(operate + '_rd')out_rd.save(savename, quality=100)# logger.info(operate + '_rd')# 切割中心x_c = int(x + (w - ww) / 2)y_c = int(y + (h - hh) / 2)out_c = image.crop((x_c, y_c, ww, hh))savename = self.get_savename(operate + '_c')out_c.save(savename, quality=100)# logger.info('提取中心')except Exception as e:logger.error('ERROR %s', operate)logger.error(e)def test():# 源地址和输出地址rootPath = 'E:/face/1.jpg'export_path_base = 'E:/face/image/'# 声明类对象imgPre = ImgPre(rootPath, export_path_base)imgPre.deform()imgPre.transpose()imgPre.rotate(15)imgPre.rotate(30)imgPre.crop()imgPre.lightness(1.07)imgPre.lightness(0.87)if __name__ == '__main__':import datetimeprint('start...')# 计时start_time = datetime.datetime.now()test()end_time = datetime.datetime.now()time_consume = (end_time - start_time).microseconds / 1000000logger.info('start_time: %s', start_time)logger.info('end_time: %s', end_time)logger.info('time_consume: %s(s)', time_consume)  # 0.280654(s)logger.info('main finish')