深度学习图像数据增强：翻转、旋转、拉伸、五部分提取、明暗度变化python

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

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

一：日志依赖

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

 

数据操作完整源码自 [email protected] 申请

二：代码实现

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 savename

        except Exception as e:
            logger.error('get_savename ERROR')
            logger.error(e)

    def lightness(self, light):
        """改变图像亮度.
        推荐值：
            0.87，1.07
        明亮程度
            darker < 1.0