《一种图像亮度和对比度的自动调节方法.pdf》由会员分享,可在线阅读,更多相关《一种图像亮度和对比度的自动调节方法.pdf(11页完整版)》请在专利查询网上搜索。
1、(10)申请公布号 CN 104036474 A (43)申请公布日 2014.09.10 CN 104036474 A (21)申请号 201410260186.4 (22)申请日 2014.06.12 G06T 5/40(2006.01) G06K 9/62(2006.01) (71)申请人 厦门美图之家科技有限公司 地址 361008 福建省厦门市湖里区火炬高新 区火炬园火炬大厦 N202 室 (72)发明人 张伟 傅松林 王喆 李志阳 (54) 发明名称 一种图像亮度和对比度的自动调节方法 (57) 摘要 本发明公开了一种图像亮度和对比度的自动 调节方法, 其通过收集样本图像对其进行亮。
2、度和 对比度调整以及人工进行环境分类, 并将人工分 类后的样本图像投入卷积神经网络里进行分类模 型训练, 然后即可使用训练好的最佳分类模型对 待处理图像进行环境分类, 并根据待处理图像的 环境类型将待处理图像与同一环境类型中的样本 图像数据库进行相似性分析, 最后将相似性最高 的样本图像的亮度和对比度调整方案作用于待处 理图像, 得到最终的结果图像, 从而实现图像亮度 和对比度的自动调节, 且效果好, 无需再次手动调 整。 (51)Int.Cl. 权利要求书 2 页 说明书 7 页 附图 1 页 (19)中华人民共和国国家知识产权局 (12)发明专利申请 权利要求书2页 说明书7页 附图1页 。
3、(10)申请公布号 CN 104036474 A CN 104036474 A 1/2 页 2 1. 一种图像亮度和对比度的自动调节方法, 其特征在于, 包括以下步骤 : 10. 收集样本图像, 并对样本图像进行最适合的亮度和对比度调整并记录调整的数 值 ; 20. 根据样本图像的亮度直方图统计结果人工地进行环境分类 ; 30. 将人工分类后的样本图像进行归一化处理, 并将处理后的样本图像投入卷积神经 网络里进行分类模型的训练 ; 40. 根据步骤 30 训练后的模型对样本图像进行环境分类, 将分类结果与人工分类不同 的定为分类错误的样本图像, 并收集起来再次执行步骤 20 至步骤 30, 直。
4、到超过预期结果时 定为所述步骤 30 所训练出来的分类模型为最佳分类模型 ; 50. 接收待处理图像, 并使用训练好的最佳分类模型对该待处理图像进行环境分类, 得 到待处理图像的环境类型 ; 60. 将待处理图像与同一环境类型中的样本图像数据库进行相似性分析, 得到相似性 最高的样本图像, 并获取该样本图像对应的亮度和对比度调整数值 ; 70. 将所述的亮度和对比度调整数值作用于待处理图像, 得到最终的结果图像。 2. 根据权利要求 1 所述的一种图像亮度和对比度的自动调节方法, 其特征在于 : 所述 的环境分类的类型包括 : 阴影类型、 高光类型、 中间类型、 阴影高光类型、 阴影中间类型、。
5、 中 间高光类型、 平衡类型。 3. 根据权利要求 1 所述的一种图像亮度和对比度的自动调节方法, 其特征在于 : 所述 的步骤 30 中将人工分类后的样本图像进行归一化处理, 主要是对所有样本图像进行直方 图统计, 得到所有类型图像的平均直方图数据, 然后再将所有样本图像的直方图都进行减 去所述的平均直方图数据得到归一化后的样本图像。 4. 根据权利要求 1 所述的一种图像亮度和对比度的自动调节方法, 其特征在于 : 所述 的步骤 30 中将归一化后的样本图像投入卷积神经网络里进行分类模型的训练。 5. 根据权利要求 4 所述的一种图像亮度和对比度的自动调节方法, 其特征在于 : 所述 的步。
6、骤 40 中根据步骤 30 训练后的模型对样本图像进行环境分类, 将分类结果与人工分类 不同的定为分类错误的样本图像, 主要是将获取的环境分类标签与人工环境分类的标签不 一致的样本图像收集起来重新进行分类, 再将重新分类后的样本图像再次进行训练学习, 重复上述过程直到分类正确为止。 6. 根据权利要求 4 所述的一种图像亮度和对比度的自动调节方法, 其特征在于 : 所述 的步骤 40 中根据步骤 30 训练后的模型对样本图像进行环境分类, 主要是将样本图像放入 卷积神经网络系统中进行计算该样本图像每一种环境类型的概率, 并选择概率最大的环境 类型作为该样本图像的环境类型。 7. 根据权利要求 。
7、1 所述的一种图像亮度和对比度的自动调节方法, 其特征在于 : 所述 的步骤 50 中将待处理图像与同一环境类型中的样本图像数据库进行相似性分析, 主要是 通过计算待处理图像的直方图统计结果与所述数据库中的样本图像的直方图统计结果的 欧式距离, 得到与待处理图像的欧式距离最短的样本图像作为相似性最高的样本图像。 8. 根据权利要求 1 或 3 或 7 所述的一种图像亮度和对比度的自动调节方法, 其特征在 于 : 所述的直方图统计是以图像的灰度值作为横坐标的值, 图像中该灰度值出现的像素个 数作为纵坐标的值。 权 利 要 求 书 CN 104036474 A 2 2/2 页 3 9. 根据权利要。
8、求 1 所述的一种图像亮度和对比度的自动调节方法, 其特征在于 : 所述 的亮度和对比度的调整公式为 : 21. 计算对比度的调整 : c (100+contrast)/100.0f ; 其中, contrast 为对比度的调整数值, 范围从 -100 到 100 ; 22. 计算亮度的调整 : brightness+ 128 ; 其中, brightness 为亮度的调整数值, 范围从 -128 到 128 ; 23. 创建亮度和对比度的调整映射表 cTable256 ; cTablei max(0, min(255, (i-128)*c+brightness+0.5) ; 其中 i 的范围。
9、从 0 到 255。 24. 对待处理图像的每个像素点的红、 绿、 蓝通道的颜色值进行数值的映射, 即 : Color cTableColor ; 其中, Color 为待处理图像的每个像素点的红、 绿、 蓝通道的颜色值。 权 利 要 求 书 CN 104036474 A 3 1/7 页 4 一种图像亮度和对比度的自动调节方法 技术领域 0001 本发明涉及一种图像处理方法, 特别是一种图像亮度和对比度的自动调节方法。 背景技术 0002 目前图像处理的自动亮度和对比度是利用图像中的各个通道的直方图统计完的 结果进行自动化调整的, 但实际上还是会有一些图亮度不足, 或者图像亮度过曝而自动亮 度。
10、和对比度后都过曝, 这些情况都需要用户手动调整亮度和对比度才能够达到更好的效 果, 给用户带来不便。 发明内容 0003 本发明为解决上述问题, 提供了一种图像亮度和对比度的自动调节方法, 其不仅 实现自动调节, 而且效果好, 无需再次手动调整。 0004 为实现上述目的, 本发明采用的技术方案为 : 0005 一种图像亮度和对比度的自动调节方法, 其特征在于, 包括以下步骤 : 0006 10. 收集样本图像, 并对样本图像进行最适合的亮度和对比度调整并记录调整的 数值 ; 0007 20. 根据样本图像的亮度直方图统计结果人工地进行环境分类 ; 0008 30. 将人工分类后的样本图像进行。
11、归一化处理, 并将处理后的样本图像投入卷积 神经网络里进行分类模型的训练 ; 0009 40. 根据步骤 30 训练后的模型对样本图像进行环境分类, 将分类结果与人工分类 不同的定为分类错误的样本图像, 并收集起来再次执行步骤 20 至步骤 30, 直到超过预期结 果时定为所述步骤 30 所训练出来的分类模型为最佳分类模型 ; 0010 50. 接收待处理图像, 并使用训练好的最佳分类模型对该待处理图像进行环境分 类, 得到待处理图像的环境类型 ; 0011 60. 将待处理图像与同一环境类型中的样本图像数据库进行相似性分析, 得到相 似性最高的样本图像, 并获取该样本图像对应的亮度和对比度调。
12、整数值 ; 0012 70. 将所述的亮度和对比度调整数值作用于待处理图像, 得到最终的结果图像。 0013 优选的, 所述的环境分类的类型包括 : 阴影类型、 高光类型、 中间类型、 阴影高光类 型、 阴影中间类型、 中间高光类型、 平衡类型。 0014 优选的, 所述的步骤 30 中将分类后的样本图像进行归一化处理, 主要是对所有样 本图像进行直方图统计, 得到所有类型图像的平均直方图数据, 然后再将所有样本图像的 直方图都进行减去所述的平均直方图数据得到归一化后的样本图像。 0015 优选的, 所述的步骤 30 中将归一化后的样本图像投入卷积神经网络里进行分类 模型训练。 0016 优选。
13、的, 所述的步骤40中根据步骤30训练后的模型对样本图像进行环境分类, 将 分类结果与人工分类不同的定为分类错误的样本图像, 主要是将获取的环境分类标签与人 说 明 书 CN 104036474 A 4 2/7 页 5 工环境分类的标签不一致的样本图像收集起来重新进行分类, 再将重新分类后的样本图像 再次进行训练学习, 重复上述过程直到分类正确为止。 0017 优选的, 所述的步骤40中根据步骤30训练后的模型对样本图像进行环境分类, 主 要是将样本图像放入卷积神经网络系统中进行计算该样本图像每一种环境类型的概率, 并 选择概率最大的环境类型作为该样本图像的环境类型。 0018 优选的, 所述。
14、的步骤 50 中将待处理图像与同一环境类型中的样本图像数据库进 行相似性分析, 主要是通过计算待处理图像的直方图统计结果与所述数据库中的样本图像 的直方图统计结果的欧式距离, 得到与待处理图像的欧式距离最短的样本图像作为相似性 最高的样本图像。 0019 优选的, 所述的直方图统计是以图像的灰度值作为横坐标的值, 图像中该灰度值 出现的像素个数作为纵坐标的值。 0020 优选的, 所述的亮度和对比度的调整公式为 : 0021 21. 计算对比度的调整 : 0022 c (100+contrast)/100.0f ; 0023 其中, contrast 为对比度的调整数值, 范围从 -100 到。
15、 100 ; 0024 22. 计算亮度的调整 : 0025 brightness+ 128 ; 0026 其中, brightness 为亮度的调整数值, 范围从 -128 到 128 ; 0027 23. 创建亮度和对比度的调整映射表 cTable256 ; 0028 cTablei max(0, min(255, (i-128)*c+brightness+0.5) ; 0029 其中 i 的范围从 0 到 255。 0030 24. 对待处理图像的每个像素点的红、 绿、 蓝通道的颜色值进行数值的映射, 即 : 0031 Color cTableColor ; 0032 其中, Color。
16、 为待处理图像的每个像素点的红、 绿、 蓝通道的颜色值。 0033 本发明的有益效果是 : 0034 本发明的一种图像亮度和对比度的自动调节方法, 其通过收集样本图像对其进行 亮度和对比度调整以及人工进行环境分类, 并将人工分类后的样本图像投入卷积神经网络 里进行分类模型训练, 然后即可使用训练好的最佳分类模型对待处理图像进行环境分类, 并根据待处理图像的环境类型将待处理图像与同一环境类型中的样本图像数据库进行相 似性分析, 最后将相似性最高的样本图像的亮度和对比度调整方案作用于待处理图像, 得 到最终的结果图像, 从而实现图像亮度和对比度的自动调节, 且效果好, 无需再次手动调 整。 附图说。
17、明 0035 此处所说明的附图用来提供对本发明的进一步理解, 构成本发明的一部分, 本发 明的示意性实施例及其说明用于解释本发明, 并不构成对本发明的不当限定。在附图中 : 0036 图 1 为本发明一种图像亮度和对比度的自动调节方法的流程简图。 具体实施方式 说 明 书 CN 104036474 A 5 3/7 页 6 0037 为了使本发明所要解决的技术问题、 技术方案及有益效果更加清楚、 明白, 以下结 合附图及实施例对本发明进行进一步详细说明。应当理解, 此处所描述的具体实施例仅用 以解释本发明, 并不用于限定本发明。 0038 如图 1 所示, 本发明的一种图像亮度和对比度的自动调节。
18、方法, 其包括以下步骤 : 0039 10. 收集样本图像, 并对样本图像进行最适合的亮度和对比度调整并记录调整的 数值 ; 0040 20. 根据样本图像的亮度直方图统计结果人工地进行环境分类 ; 0041 30. 将人工分类后的样本图像进行归一化处理, 并将处理后的样本图像投入卷积 神经网络里进行分类模型的训练 ; 0042 40. 根据步骤 30 训练后的模型对样本图像进行环境分类, 将分类结果与人工分类 不同的定为分类错误的样本图像, 并收集起来再次执行步骤 20 至步骤 30, 直到超过预期结 果时定为所述步骤 30 所训练出来的分类模型为最佳分类模型 ; 这里的预期结果是指亮度 环。
19、境检测的准确率达到预设值, 本实施例中优选的预设值为 90。 0043 50. 接收待处理图像, 并使用训练好的最佳分类模型对该待处理图像进行环境分 类, 得到待处理图像的环境类型 ; 0044 60. 将待处理图像与同一环境类型中的样本图像数据库进行相似性分析, 得到相 似性最高的样本图像, 并获取该样本图像对应的亮度和对比度调整数值 ; 0045 70. 将所述的亮度和对比度调整数值作用于待处理图像, 得到最终的结果图像。 0046 本实施例中, 所述的环境分类的类型包括 : 阴影类型、 高光类型、 中间类型、 阴影高 光类型、 阴影中间类型、 中间高光类型、 平衡类型 ; 所述的阴影类型。
20、表示直方图中阴影的数 据最多, 中间、 高光的数据接近并且偏低 ; 所述的高光类型表示直方图中高光的数据最多, 阴影、 中间的数据接近并且偏低 ; 所述的中间类型表示直方图中中间的数据最多, 阴影、 高 光的数据接近并且偏低 ; 所述的阴影高光类型表示直方图中阴影、 高光的数据接近并且最 多, 中间的数据偏低 ; 所述的阴影中间类型表示直方图中阴影、 中间的数据接近并且最多, 高光的数据偏低 ; 所述的高光中间类型表示直方图中高光、 中间的数据接近并且最多, 阴影 的数据偏低 ; 所述的平衡类型表示直方图中高光、 中间、 阴影的数据平均。 0047 所述的步骤 30 中将分类后的样本图像进行归。
21、一化处理, 主要是对所有样本图像 进行直方图统计, 得到所有类型图像的平均直方图数据, 然后再将所有样本图像的直方图 都进行减去所述的平均直方图数据得到归一化后的样本图像 ; 所述的步骤 30 中将处理后 的样本图像投入卷积神经网络里进行分类模型训练, 主要是对所有样本图像进行采样, 得 到所有样本图像在分类模型里的环境分类标签。 0048 所述的步骤 40 中根据步骤 30 训练后的模型对样本图像进行环境分类, 将分类结 果与人工分类不同的定为分类错误的样本图像, 主要是将获取的环境分类标签与人工环境 分类的标签不一致的样本图像收集起来重新进行分类, 即, 调整网络结构, 再将重新分类后 的。
22、样本图像再次进行训练学习, 如此重复 “训练 - 调整网络结构 - 再训练” 的过程直到分 类正确为止。 0049 本实施例中网格结构顺序为输入层 -K 个小组层 - 全连接层 -SoftMax 层, 其中 K 大于等于 1 ; 小组层包括卷积层、 激活层、 下采样层、 归一化层 ; 卷积层、 激活层、 下采样层、 归一化层中每个层的核大小以及输出大小都是可以进行任意调节的, 并且每个层都有一个 说 明 书 CN 104036474 A 6 4/7 页 7 输入且产生一个输出, 每一层的输出作为下一层的输入。 0050 其中, 输入层的输入大小为 Height x Weight x Chann。
23、el, 其中 Weight、 Height 为输 入层图像的宽和高, Channel 为输入层图像的颜色通道 ; 由于本发明使用 GPU 硬件实现的原 因, Weight Height ; 输入图像的 channel 只能为 1 或者 3。 0051 卷积层 : 0052 1) 核的大小必须是奇数, 且不大于该层输入的宽或者高 ; 0053 2) 中间表示通过卷积层时不改变宽和高, 通道数可变可不变 ; 理论上可以为任意 正整数, 由于本发明使用 GPU 硬件实现的原因, 这里为 16 的倍数。 0054 激活层 : 0055 1) 激活层不改变卷积层表示的宽、 高或者通道数 ; 0056 2。
24、) 激活层所使用的激活函数包括但不限于以下函数类型 : 0057 f(x) 1/(1+e-x) 0058 f(x) a*tanh(b*x), a, b 为任意非零实数 0059 f(x) max(0, x) 0060 f(x) min(a, max(0, x) 0061 f(x) log(1+ex) 0062 f(x) |x| 0063 f(x) x2 0064 0065 f(x) ax+b 0066 3) 激活层跟在卷积层或者全连接之后。 0067 下采样层 : 0068 1) 下采样层不改变中间表示的通道数 ; 0069 2) 下采样层对图像的缩小比即为核的大小 : 即核为 m x n 的。
25、下采样层会造成中间 表示缩小为上一层的 (1/m)x(1/n), 理论上 m 和 n 可为任意自然数, 由于本发明使用 GPU 硬 件实现的原因, m n。例如, 15x15x32 通过 3x3 的下采样后, 变成 5x5x32 ; 15x15x32 通过 5x5 的下采样后, 变成 3x3x32 ; 但是 15x15x32 不能进行 2x2 的下采样, 因为 15 不能被 2 整 除 ; 并不是说, 输入尺寸必须是 2 的次幂, 即 16、 32、 64 等, 输入尺寸只要保证能被所有下采 样层采样即可。 0070 归一化层 : 0071 1) 归一化层不改变中间表示的任何尺寸 ; 0072。
26、 2) 归一化层不是必须的, 可要可不要, 添加归一化层通常会提高精度并增加计算 量 ; 是否添加归一化层, 要看添加后实际提升的精度和损失的速度。 0073 一般的组合是 : 卷积 - 激活 - 下采样 - 归一化。 0074 以下情况特殊 : 0075 1) 添加归一化层对精度提升较小却增大了很多运算量时, 取消归一化层, 即采用 以下组合 : 卷积 - 激活 - 下采样 ; 0076 2) 归一化层提前, 效果基本相同, 即采用以下组合 : 卷积 - 激活 - 归一化 - 下 说 明 书 CN 104036474 A 7 5/7 页 8 采样。 0077 3) 取消下采样层 : 卷积 -。
27、 激活 ; 或者卷积 - 激活 - 归一化 ; 下采样本质是为了 增加鲁棒性, 同时顺便有减少后续层的运算量的作用 ; 一个网络中通常会有几层下采样, 但 并不是所有的 “卷积 - 激活” 后面都要跟下采样。 0078 全连接层 : 0079 1) 通过全连接层后的中间表示会变成 1 维的, 不再是 3 维的 ; 0080 2) 全连接的输出可以任意 ; 0081 3) 一旦进过全连接, 就无法进行卷积、 下采样或归一化 ; 0082 4) 全连接后面可以接激活层, 或者继续接全连接。 0083 SoftMax 层 : 0084 接在全连接层之后, 作用是把全连接产生的实值变成 0, 1 之间。
28、的概率。 0085 本发明最后使用的网络结构如表 1 所示。 0086 表 1 卷积神经网络结构 0087 层数 类型核大小输出大小注解 1输入层 32x32x3 2卷积层5x532x32x32 3激活层 32x32x32 4下采样层2x216x16x32f(x) x2 5归一化层 16x16x32使用局部归一化 6卷积层5x516x16x16 7激活层 16x16x16 8下采样层2x28x8x16f(x) |x| 9归一化层 8x8x16使用局部归一化 10全连接层 7 个数据 11SoftMax 层 7 个数据 0088 0089 所述的步骤 40 中根据步骤 30 训练后的模型对样本图。
29、像进行环境分类, 主要是将 样本图像放入卷积神经网络系统中进行计算该样本图像每一种环境类型的概率, 并选择概 率最大的环境类型作为该样本图像的环境类型。具体主要是通过对样本图像进行采样, 得 到样本图像块并放入神经网络的输入层, 进行全连接后, 在最后的 SoftMax 层得到每一种 环境类型标签的概率, 即在区间 0, 1 中的实值 ; 本实施例中根据环境类型分为 : 阴影类 说 明 书 CN 104036474 A 8 6/7 页 9 型、 高光类型、 中间类型、 阴影高光类型、 阴影中间类型、 中间高光类型、 平衡类型, 共 7 种类 型的环境标签, 即 7 个数据, 这 7 个数据的和。
30、等于 1 ; 然后, 将得到的每个样本图像块的标签 的概率进行平均, 得到样本图像块的标签的概率, 选择概率最大的标签作为该样本图像的 环境类型的标签。 0090 所述的步骤 50 中将待处理图像与同一环境类型中的样本图像数据库进行相似性 分析, 主要是通过计算待处理图像的直方图统计结果与所述数据库中的样本图像的直方图 统计结果的欧式距离, 得到与待处理图像的欧式距离最短的样本图像作为相似性最高的样 本图像。 0091 所述的欧氏距离的计算公式如下 : 0092 d sqrt( (xi1-xi2)2) ; 0093 其中, i 1, 2n, xi1 为第一个点的第 i 维坐标, xi2 为第二。
31、个点的第 i 维坐标, d 为计算得到的欧氏距离。n 维欧氏空间是一个点集, 它的每个点可以表示为 (x(1), x(2), x(n), 其中x(i)(i1, 2n)是实数, 称为x的第i个坐标, 两个点x和y(y(1), y(2) y(n) 之间的距离 d(x, y) 定义为上述公式中的欧氏距离。 0094 步骤10及步骤50中, 所述的直方图统计是以图像的灰度值作为横坐标的值, 图像 中该灰度值出现的像素个数作为纵坐标的值。 0095 步骤 20 及步骤 60 中, , 所述的亮度和对比度的调整公式为 : 0096 21. 计算对比度的调整 : 0097 c (100+contrast)/。
32、100.0f ; 0098 其中, contrast 为对比度的调整数值, 范围从 -100 到 100 ; 0099 22. 计算亮度的调整 : 0100 brightness+ 128 ; 0101 其中, brightness 为亮度的调整数值, 范围从 -128 到 128 ; 0102 23. 创建亮度和对比度的调整映射表 cTable256 ; 0103 cTablei max(0, min(255, (i-128)*c+brightness+0.5) ; 0104 其中 i 的范围从 0 到 255。 0105 24. 对待处理图像的每个像素点的红、 绿、 蓝通道的颜色值进行数值。
33、的映射, 即 : 0106 Color cTableColor ; 0107 其中, Color 为待处理图像的每个像素点的红、 绿、 蓝通道的颜色值。 0108 本发明的图像亮度和对比度的自动调节方法, 通过收集各种环境的样本图像并根 据其亮度直方图进行分类, 并且人工调整每张样本图像最适合的亮度和对比度, 然后通过 卷积神经网络对样本图像进行模型的训练。对待处理的图像通过训练好的模型进行环境 分类, 接着对待处理图像与其环境类型里的样本图像根据直方图统计的结果进行相似性分 析, 得到相似性最高的样本图像, 并获取该样本图像对应的亮度和对比度调整数值, 将该亮 度和对比度数值作用于待处理图像。
34、, 得到最终结果图, 使得亮度和对比度自动调节的效果 更好。 0109 上述说明示出并描述了本发明的优选实施例, 应当理解本发明并非局限于本文所 披露的形式, 不应看作是对其他实施例的排除, 而可用于各种其他组合、 修改和环境, 并能 够在本文发明构想范围内, 通过上述教导或相关领域的技术或知识进行改动。而本领域人 说 明 书 CN 104036474 A 9 7/7 页 10 员所进行的改动和变化不脱离本发明的精神和范围, 则都应在本发明所附权利要求的保护 范围内。 说 明 书 CN 104036474 A 10 1/1 页 11 图 1 说 明 书 附 图 CN 104036474 A 11 。