一种基于光照变换网络的图像自动增强方法

基本信息

申请号 CN202011624087.1 申请日 -
公开(公告)号 CN112686823A 公开(公告)日 2021-04-20
申请公布号 CN112686823A 申请公布日 2021-04-20
分类号 G06N5/04(2006.01)I;G06T5/00(2006.01)I;G06K9/46(2006.01)I;G06N3/08(2006.01)I;G06K9/62(2006.01)I;G06N3/04(2006.01)I 分类 计算;推算;计数;
发明人 苏家仪;韦光亮;王筱东;张玉国;苏世宁;韦潇依;龚骏毅 申请(专利权)人 广西慧云信息技术有限公司
代理机构 南宁东之智专利代理有限公司 代理人 戴燕桃;汪治兴
地址 530007广西壮族自治区南宁市高新区创新路23号9号楼三层
法律状态 -

摘要

摘要 本发明公开一种基于光照变换网络的图像自动增强方法,包括以下步骤:(1)数据集制作:收集大量图像分类数据,并将图像分类数据按0.8:0.1:0.1的比例随机划分为训练集、验证集与测试集;(2)构建一种光照变换网络:所述光照变换网络由图像自动增强模块与ResNet50模块组成;所述光照变换网络用于对输入的原始图像进行自动增强;(3)模型训练:指定步骤(1)中的训练集与验证集,使用SGD优化器,将训练集输入至步骤(2)的光照变换网络,进行模型训练;(4)模型推理。本发明根据输入图像进行自动增强,可有效提升深度学习模型的光照一致性。