一种基于深度神经网络的颜色空间转换方法技术

本发明专利技术公开了一种基于深度神经网络的颜色空间转换方法，具体按照以下步骤实施：步骤1、制作训练样本和测试样本，训练样本用来建立神经网络模型，测试样本用来检验训练完成的模型的转换精度；建立深度置信网络模型；步骤2、使用粒子群算法优化深度置信网络的神经元连接权值等参数；步骤3、将训练样本输入到步骤3进行训练，随后使用BP神经网络进行反向微调，得到稳定的PSO

【技术实现步骤摘要】
一种基于深度神经网络的颜色空间转换方法


[0001]本专利技术属于图像处理
，涉及一种基于深度神经网络的颜色空间转换方法。

技术介绍

[0002]数码印花机的色彩管理可以分为校准设备、特征化和颜色空间转换三个步骤，颜色空间转换是数码印花色彩管理的重要部分。CMYK颜色空间是应用在数码印花中的色彩标准，描述的是数码印花产品中青(C)、洋红(M)、黄(Y)、黑(K)四种颜色的墨量之间的关系。L*a*b*颜色空间与设备无关，可以作为不同设备之间的连接色空间，广泛应用于数码印花机的色彩评价。L*a*b*中L取值代表明亮度；a值正数范围为红色色域，负数范围为绿色色域；b值正数范围为黄色色域，负数范围为蓝色色域。不同设备对图像具有不同的颜色空间描述方法，其色域具有较大差别，导致数码印花产品与样稿图像之间存在色差。建立转换精度较高的L*a*b*与CMYK颜色空间的转换关系，能很大程度提升数码印花产品质量。
[0003]神经网络技术在色彩管理和颜色空间转换应用中受到极大关注。传统的颜色空间转换方法使用浅层神经网络，如BPNN，GRNN和ELM等，它们受其本身结构的影响，在复杂问题下很容易获得局部最优解，其精度难以进一步提高。深度置信网络(Deep Belief Network,DBN)是一种无监督的学习方法，它可以从大量数据中提取特征，并且具有广泛的适应性和强大的映射能力，适合构建颜色空间转换模型。DBN算法的参数往往需要人工凭借经验和多次调整来确定，极大影响了网络实用性。粒子群优化算法(Particle Swarm Optimization,PSO)可对DBN算法的参数进行寻优，最后将最优参数赋给DBN网络，提高DBN的转换精度。

技术实现思路

[0004]本专利技术的目的是提供一种基于深度神经网络的颜色空间转换方法，解决了现有技术中存在的L*a*b*到CMYK颜色空间转换模型转换精度低的问题。
[0005]本专利技术所采用的技术方案是，一种基于深度神经网络的颜色空间转换方法，具体按照以下步骤实施：
[0006]步骤1、制作训练样本和测试样本，训练样本用来建立神经网络模型，测试样本用来检验训练完成的模型的转换精度；建立深度置信网络模型，初始化DBN中输入层、隐含层、输出层各层之间的参数，L*a*b*颜色空间作为神经网络的输入值，CMYK颜色空间作为神经网络的输出值；
[0007]步骤2、使用粒子群算法优化深度置信网络的神经元连接权值等参数；
[0008]步骤3、将训练样本输入到步骤3进行训练，随后使用BP神经网络进行反向微调，得到稳定的PSO-DBN模型，得到L*a*b*到CMYK颜色空间转换模型；
[0009]步骤4、将测试样本输入转换模型进行颜色转换并计算转换误差，检验模型精度，完成颜色空间转换。
[0010]本专利技术的特点还在于：
[0011]步骤1中建立深度置信网络模型，初始化DBN中输入层、隐含层、输出层各层之间的参数，L*a*b*颜色空间作为神经网络的输入值，CMYK颜色空间作为神经网络的输出值具体按照以下实施：建立深度置信网络模型，受限玻尔兹曼机是DBN的主要组成部分，DBN的训练过程可以分为两个阶段，即预训练和反向微调；首先，采用无监督贪婪学习算法逐层训练网络中的每个RBM，并逐层传递数据特征信息，以实现网络参数的初始化；随后，使用BP神经网络算法进行微调，通过从顶部到底部的预训练获得的初始权重；该过程进行监督训练，使模型得到最优解，从而确定整个DBN网络的结构。
[0012]步骤1中RBM的无监督训练具体按照以下实施：
[0013]在RBM中，v1,v2,
…
表示可见层，h1,h2,
…
表示隐藏层，w
ij
表示每个神经元连接的权重；引入能量函数来定义系统的总能量，并计算联合分布概率；
[0014][0015]在上式中θ＝{a,b,w},w
ij
是可见层和隐含层之间的连接权重，V是可见层中的单元数，H是隐含中的单元数，a
i
是可见层的偏移量，a
j
是隐藏层的偏移量；根据定义的系统能量，可见和隐藏单元的联合分布定义如下。
[0016][0017]Z＝∑
v
′
∑
h
′
e-E(v
′
,h
′
|θ)
ꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀ3[0018]在上式中，Z是归一化因子，保证联合概率在[0,1]的范围内变化。因此，隐藏层的边缘分布如下：
[0019][0020]由于RBM每一层中的神经元都是独立的，因此可以根据以下关系式获得每个单元的激活概率；
[0021]p(v
i
＝I|h)＝f(a
i
+∑
j
h
j
w
ij
)
ꢀꢀꢀꢀꢀꢀꢀꢀ5[0022]p(h
i
＝I|v)＝f(b
j
+∑
i
v
i
w
ij
)
ꢀꢀꢀꢀꢀꢀꢀꢀ
6。
[0023]步骤1中DBN网络的训练过程具体按照以下实施：
[0024]步骤1.1，设结点状态表示为F
i
，F
j
表示与结点i相连的结点j的状态，权重矩阵为W；随机选取一个训练样本，将其数据输入到第一个RBM的可见层，依公式(7)更新第一层隐含层各结点的状态F
j
，其中σ∈[0,1]。
[0025][0026]步骤1.2，利用步骤2.1求出的隐含结点状态F
j
，依据公式(8)更新第一个RBM可见结点状态，记为F'
i
；
[0027][0028]步骤1.3，将上一步求得的第一个RBM的隐含层结点状态F'
i
作为第二个RBM的输入，再按照上述步骤逐层更新DBN网络的各结点状态，直至将四个RBM结点更新完毕；
[0029]步骤1.4，根据公式(9)计算网络权重值，更新网络中的权重矩阵，直到网络的权重
矩阵的变化量足够小或达到设定的最高训练次数，结束DBN训练；
[0030]Δω
ij
＝η(＜F
i
F
j
＞-＜F'
i
F'
j
＞)
ꢀꢀꢀꢀꢀꢀꢀꢀ
9。
[0031]步骤2具体按照以下步骤实施：
[0032]步骤2.1，对数据集进行预处理，首先初始化DBN神经网络中的参数，以此确定粒子的维度；
[0033]步骤2.2，初始化粒子群各参数，对4层隐藏层DBN，每层分别有m1、m2、m3和m4个神经元，学习率η∈[0,1)；因此设定粒子群中的每一个粒子为一个四维向量X(m1,m2,m3,m4,η)；
[0034]步骤2.3，利用公式(10)本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于深度神经网络的颜色空间转换方法，其特征在于，具体按照以下步骤实施：步骤1、制作训练样本和测试样本，训练样本用来建立神经网络模型，测试样本用来检验训练完成的模型的转换精度；建立深度置信网络模型，初始化DBN中输入层、隐含层、输出层各层之间的参数，L*a*b*颜色空间作为神经网络的输入值，CMYK颜色空间作为神经网络的输出值；步骤2、使用粒子群算法优化深度置信网络的神经元连接权值等参数；步骤3、将训练样本输入到步骤3进行训练，随后使用BP神经网络进行反向微调，得到稳定的PSO-DBN模型，得到L*a*b*到CMYK颜色空间转换模型；步骤4、将测试样本输入转换模型进行颜色转换并计算转换误差，检验模型精度，完成颜色空间转换。2.根据权利要求1所述的一种基于深度神经网络的颜色空间转换方法，其特征在于，所述步骤1中建立深度置信网络模型，初始化DBN中输入层、隐含层、输出层各层之间的参数，L*a*b*颜色空间作为神经网络的输入值，CMYK颜色空间作为神经网络的输出值具体按照以下实施：建立深度置信网络模型，受限玻尔兹曼机是DBN的主要组成部分，DBN的训练过程可以分为两个阶段，即预训练和反向微调；首先，采用无监督贪婪学习算法逐层训练网络中的每个RBM，并逐层传递数据特征信息，以实现网络参数的初始化；随后，使用BP神经网络算法进行微调，通过从顶部到底部的预训练获得的初始权重；该过程进行监督训练，使模型得到最优解，从而确定整个DBN网络的结构。3.根据权利要求2所述的一种基于深度神经网络的颜色空间转换方法，，其特征在于，所述步骤1中RBM的无监督训练具体按照以下实施：在RBM中，v1,v2,
…
表示可见层，h1,h2,
…
表示隐藏层，w
ij
表示每个神经元连接的权重；引入能量函数来定义系统的总能量，并计算联合分布概率；在上式中θ＝{a,b,w},w
ij
是可见层和隐含层之间的连接权重，V是可见层中的单元数，H是隐含中的单元数，a
i
是可见层的偏移量，a
j
是隐藏层的偏移量；根据定义的系统能量，可见和隐藏单元的联合分布定义如下。Z＝∑
v
′
∑
h
′
e-E(v
′
,h
′
|θ)
ꢀꢀꢀꢀꢀꢀꢀꢀ
(3)在上式中，Z是归一化因子，保证联合概率在[0,1]的范围内变化。因此，隐藏层的边缘分布如下：由于RBM每一层中的神经元都是独立的，因此可以根据以下关系式获得每个单元的激活概率；p(v
i
＝I|h)＝f(a
i
+∑
j
h
j
w
ij
)
ꢀꢀꢀꢀꢀꢀꢀ
(5)p(h
i
＝I|v)＝f(b
j
+∑
i
v
...

【专利技术属性】
技术研发人员：苏泽斌，杨金锴，李鹏飞，景军锋，张缓缓，
申请(专利权)人：西安工程大学，
类型：发明
国别省市：