CN113935899A - 一种基于语义信息和梯度监督的船牌图像超分辨方法 - Google Patents
一种基于语义信息和梯度监督的船牌图像超分辨方法 Download PDFInfo
- Publication number
- CN113935899A CN113935899A CN202111038079.3A CN202111038079A CN113935899A CN 113935899 A CN113935899 A CN 113935899A CN 202111038079 A CN202111038079 A CN 202111038079A CN 113935899 A CN113935899 A CN 113935899A
- Authority
- CN
- China
- Prior art keywords
- image
- network
- ship plate
- noise
- resolution
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 238000000034 method Methods 0.000 title claims abstract description 46
- 230000006870 function Effects 0.000 claims abstract description 64
- 238000012549 training Methods 0.000 claims abstract description 62
- 238000000605 extraction Methods 0.000 claims abstract description 24
- 238000007781 pre-processing Methods 0.000 claims abstract description 3
- 238000004364 calculation method Methods 0.000 claims description 22
- 238000013507 mapping Methods 0.000 claims description 20
- 230000004913 activation Effects 0.000 claims description 19
- 230000008447 perception Effects 0.000 claims description 14
- 238000010276 construction Methods 0.000 claims description 12
- 238000005070 sampling Methods 0.000 claims description 12
- 230000006835 compression Effects 0.000 claims description 11
- 238000007906 compression Methods 0.000 claims description 11
- 230000002457 bidirectional effect Effects 0.000 claims description 7
- 230000006872 improvement Effects 0.000 claims description 5
- 238000012360 testing method Methods 0.000 claims description 5
- 238000013519 translation Methods 0.000 claims description 5
- 238000010606 normalization Methods 0.000 claims description 4
- 239000013598 vector Substances 0.000 claims description 4
- 238000012935 Averaging Methods 0.000 claims description 3
- 238000005286 illumination Methods 0.000 claims description 3
- 238000005457 optimization Methods 0.000 claims description 3
- 230000008569 process Effects 0.000 claims description 3
- 230000003042 antagnostic effect Effects 0.000 claims description 2
- 238000013528 artificial neural network Methods 0.000 claims description 2
- 230000008034 disappearance Effects 0.000 claims description 2
- 238000012795 verification Methods 0.000 claims description 2
- 230000000694 effects Effects 0.000 description 8
- 238000013135 deep learning Methods 0.000 description 6
- 230000008901 benefit Effects 0.000 description 5
- 238000012545 processing Methods 0.000 description 4
- XLYOFNOQVPJJNP-UHFFFAOYSA-N water Substances O XLYOFNOQVPJJNP-UHFFFAOYSA-N 0.000 description 4
- 230000009286 beneficial effect Effects 0.000 description 3
- 238000012544 monitoring process Methods 0.000 description 3
- ORILYTVJVMAKLC-UHFFFAOYSA-N Adamantane Natural products C1C(C2)CC3CC1CC2C3 ORILYTVJVMAKLC-UHFFFAOYSA-N 0.000 description 2
- 230000003321 amplification Effects 0.000 description 2
- 238000005034 decoration Methods 0.000 description 2
- 230000007547 defect Effects 0.000 description 2
- 238000011161 development Methods 0.000 description 2
- 238000010586 diagram Methods 0.000 description 2
- 238000012986 modification Methods 0.000 description 2
- 230000004048 modification Effects 0.000 description 2
- 238000003199 nucleic acid amplification method Methods 0.000 description 2
- 230000003044 adaptive effect Effects 0.000 description 1
- 230000008485 antagonism Effects 0.000 description 1
- 230000005540 biological transmission Effects 0.000 description 1
- 238000013527 convolutional neural network Methods 0.000 description 1
- 230000002950 deficient Effects 0.000 description 1
- 238000005516 engineering process Methods 0.000 description 1
- 230000002708 enhancing effect Effects 0.000 description 1
- 238000011835 investigation Methods 0.000 description 1
- 238000011056 performance test Methods 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 230000007306 turnover Effects 0.000 description 1
- 230000000007 visual effect Effects 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T3/00—Geometric image transformations in the plane of the image
- G06T3/40—Scaling of whole images or parts thereof, e.g. expanding or contracting
- G06T3/4053—Scaling of whole images or parts thereof, e.g. expanding or contracting based on super-resolution, i.e. the output image resolution being higher than the sensor resolution
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/044—Recurrent networks, e.g. Hopfield networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Artificial Intelligence (AREA)
- Biomedical Technology (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- Life Sciences & Earth Sciences (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Health & Medical Sciences (AREA)
- Image Processing (AREA)
- Image Analysis (AREA)
Abstract
本发明公开了一种基于语义信息和梯度监督的船牌图像超分辨方法。本发明包括:1、收集船舶原始图像并进行预处理。2、针对于船牌文字,构建特征提取网络和超分辨重建网络,形成生成网络模型。然后通过DF2K数据集进行对抗学习预训练,得到生成网络预训练模型。3、在生成网络预训练模型中引入BLSTM结构后采用船牌文字数据集,进行训练。4:在进行对抗学习训练时,由于船牌文字区域具有一定的锐度,因此为了更好的指导网络生成超分辨的船牌文字,增加文字梯度损失函数以加强对生成网络的监督。本发明能够明显提升船牌文字的清晰程度,帮助人工快速的辨识船牌信息,方便海河航运的交通管制,提高水路航运的安全性。
Description
技术领域
本发明属于深度学习、图像处理、航运河道船只智能管理与监控、超分辨重建的技术领域,涉及一种基于生成对抗网络,以语义信息和梯度监督为指导的船牌图像超分辨率方法。
背景技术
中国航运水路运输***发达,内河网络结构完善,港口分布稠密,极大地推动了货船航运产业的发展。内河航运已然成为现代综合运输体系的重要组成部分,是水资源合理开发和综合利用的主要内容之一。据统计,仅2020年10月份全国水路货物运输总量就达到了70,659万吨,货物周转量达到92,539,819万吨。因此保证船只行驶规范、高效和安全,对于打造全信息化、智能化内河航运***具有重要意义,是构建智慧型水上“高速公路”的基础。
对船舶的监管,目前主要靠通过运河两岸、码头、港口等地安装监控摄像头,通过拍摄船只图像(主要是船只上悬挂的船牌文字图像)进行监管。然而在实际获取船牌文字的场景中,由于船舶距离摄像头过远、空气质量较差、硬件设备老旧、图像传输等因素,常常会造成摄像头采集到的船牌文字图像变得模糊,模糊的船牌文字图像不利于进一步的船牌文字识别工作,也不利于作为船舶违规行驶的证据保留,更不利于船舶在交管部门监管下的安全行驶。因此急需一种解决船牌文字图像模糊、提升船牌文字分辨率的方案。
目前主要的解决方案有基于传统方法和基于深度学习两种,但这些方案在船牌文字超分辨中又存在着诸多问题:1)传统的图像超分辨方法由于算法相对简单,在模糊船牌文字上的表现效果不佳。2)深度学习方法中,用双三次线性插值构建训练数据集的方法不能满足实际场景的应用。3)深度学习的超分辨方法缺乏对船牌文字区域语义特征的研究,使得算法针对文字超分辨的效果不理想。4)现有的深度学习方法在处理文字锐化效果方面仍有欠缺。因此,目前依然缺乏针对船牌文字超分辨方法。
发明内容
本发明的目的是针对现有技术的不足,提供一种基于语义信息和梯度监督的船牌图像超分辨方法。
本发明采取的技术方案包含以下步骤:
步骤1:收集船舶原始图像并进行预处理。
首先拍摄河道中原始船舶图像(含船牌文字),并进行人工标注船牌文字位置,裁剪得到高分辨率船牌文字区域图像(以下简称HR文字图像)。随后对HR文字图像进行降采样操作并加入噪声,生成低分辨率船牌文字区域图像(以下简称LR文字图像),完成船牌文字数据集的构造。
步骤2:针对于船牌文字,构建特征提取网络和超分辨重建网络,形成生成网络模型。然后通过DF2K数据集进行对抗学习预训练,在判别器的指导下,通过对抗学习对生成网络模型参数进行预训练,得到生成网络预训练模型。
步骤3:在生成网络预训练模型基础上引入双向LSTM(BLSTM)结构,增强对船牌文字区域的语义特征提取;采用船牌文字数据集,进行训练。
步骤4:在进行对抗学习训练时,由于船牌文字区域具有一定的锐度,因此为了更好的指导网络生成超分辨的船牌文字,增加文字梯度损失函数以加强对生成网络的监督。
所述步骤1,具体操作如下:
1-1、根据不同光照、距离、清晰(模糊)程度,在河道中拍摄船舶的高清图像和真实模糊图像,以增强数据集分布的多样性。然后对其中的高清图像和模糊图像中的船牌文字区域进行剪裁,对应得到HR文字图像和真实低分辨船牌文字图像,其中真实低分辨船牌文字图像作为测试集在训练完成后进行模型验证。
1-2、对于提取船牌文字图像原生噪声部分,根据设定的船牌文字噪声patch大小,对噪声patch的方差和均值进行限制,选取方差和均值均在设定阈值区间内的噪声patch区域并保存,遍历所有HR文字图像后得到噪声patch池。
船牌文字图像中噪声patch的均值、方差计算公式如下:
其中xp代表噪声patch中的第p个像素值,p代表每个噪声patch中的像素总个数,M代表噪声patch的均值,s2代表噪声patch的方差。
由于需要构建低分辨船牌文字图像,使用双三次插值(bicubic)的方式,对HR文字图像进行缩放,形成初步人工模糊图像。降采样公式如下:
ID=(IHR*Kbic)↓s (3)
其中,ID表示初步人工模糊图像,IHR表示HR文字图像,Kbic表示双三次插值核,s表示降采样倍数。
1-3、对初步人工模糊图像添加噪声。噪声的来源有3个方面,包括高斯噪声、JPEG压缩噪声和步骤1-2提取到的真实噪声。依次按照高斯噪声、JPEG压缩噪声和真实噪声的顺序,对初步人工模糊图像进行噪声添加,获取最终的LR文字图像,此时便完成对船牌文字数据集的构建。高斯噪声的概率密度计算方式如下:
其中x为随机变量,p(x)为概率密度,δ为标准差,μ为均值。依次添加高斯噪声、JPEG压缩噪声和真实噪声patch的公式如下:
ILR=ID+ng+njpeg+ni,i∈{1,2,...,m} (5)
其中,ILR表示LR文字图像,ID表示初步人工模糊图像,ng表示高斯噪声,njpeg表示JPEG压缩噪声,ni表示从噪声patch池中随机提取的一个噪声patch。m表示噪声patch池中的噪声patch的数量。
所述步骤2,具体操作如下:
2-1、针对于船牌文字,构建输入数据形状为(N,C,H,W)的特征提取网络和超分辨重建网络,形成生成网络模型。因为最终是为了训练船牌数据,所以此处直接针对船牌训练设置有网络输入形状。
在特征提取部分,为保证将来生成图像的多样性和细节丰富,网络中不使用标准的BN层。网络首先采用标准卷积层,对输入的DF2K数据集中的低分辨图像特征进行简单提取。然后通过一系列的卷积模块,搭建足够深的神经网络对图像的抽象特征进行提取,为保证浅层网络特征高效传播和避免梯度消失,本方法采用了残差连接和密集连接的方式,对深层网络进行优化。残差连接的计算公式如下:
xl=F(xl-1)+xl-1 (6)
其中,xl-1表示残差单元的输入特征,F(xl-1)表示学习到的加权后的残差映射,xl表示残差单元的输出特征。密集连接的计算公式如下:
xl=F([x1,x2,...,xl-1]) (7)
其中,x1,x2,...,xl-1表示密集连接单元的各个输入特征,F([x1,x2,...,xl-1])表示密集连接的映射函数,xl表示密集连接单元的输出特征。
最后,通过插值、卷积并激活的操作,对特征提取后的图像进行超分辨率重建,得到超分辨率重建图像。卷积并激活的公式如下:
其中x为输入特征,y为激活后的输出特征,w为卷积层权重,b为卷积层偏置,relu为激活函数。
2-2、通过DF2K数据集进行对抗学习预训练。
在判别网络中,输入生成图像(超分辨率重建图像)和真实图像(DF2K数据集中的高分辨图像),通过对抗学习对生成网络模型参数进行预训练,最终得到一个参数初始化良好的生成网络预训练模型。这部分主要包括判别网络的搭建和损失函数的构建。
在搭建判别网络时,使用卷积层、BN层和线性映射层作为判别网络的主要模块。在卷积层的搭建中,使用较大的卷积核,为了能够使网络获得较大感受野,提取到图像的全局的信息,使得判别网络更能够从图像的整体角度判别图像的真实性。每层卷积后再进行数据归一化操作,即加入BN层,这种操作能够稳定网络训练,增强鲁棒性。数据归一化的公式如下:
其中x为需要归一化的输入数据,y为归一化后的结果,mean(x)和var(x)分别为输入数据的均值和方差,eps为防止分母为0的量,γ和β为可训练参数。
最后,通过线性映射层,将图像特征的高维数据转换为向量,最终通过激活层将该向量映射成输入图像(超分辨率重建图像或DF2K数据集中的高分辨图像)为真实高分辨图像的概率。线性映射层的计算公式如下:
y=x·AT+b (10)
其中x为线性映射层的输入,AT为线性映射层的权重,b为线性映射层的偏置,y为线性映射层的输出。
Sigmoid激活函数的计算公式如下:
其中x为输入数据,f(x)为概率。
2-3、在预训练网络中,损失函数主要由L1损失(L1 loss)、感知损失(perceptualloss)和对抗损失(adversarial loss)三部分构成。总的损失函数计算公式如下:
Ltotal=λ1·L1+λper·Lper+λadv·Ladv (12)
其中,L1、Lper和Ladv分别表示L1损失、感知损失和对抗损失,λ1、λper和λadv分别表示对应损失函数的权重。
这三部分损失的详细叙述如下:
1)、L1损失衡量的是两幅图像之间的空间距离,可以计算两幅图像(HR和SR)像素级别的误差,能够增强两幅图像像素级的相似度,从而指导网络更好得生成SR图像。L1损失函数计算公式如下:
L1=mean(ι(x,y)) (13)
ι(x,y)={c1,...,cN}T,ci=|xi-yi| (14)
其中,N代表batch size,x和y代表形状相同的任意尺寸的图像,c表示两幅图像像素级别的差值,mean代表对所有元素取均值操作。
2)、感知损失通过将HR图像和SR图像分别输入预训练好的深度网络,然后提取中间层卷积后的特征图像做为感知特征。最终根据L1损失计算公式,计算HR图像和SR图像感知特征的L1距离,即可获得感知损失。
3)、对抗损失衡量的是网络判别器的对图像判别结果的误差。在训练时,每次计算生成对抗损失函数,分别向判别器输入一个batch的真实图像和生成图像,判断输入图像是真实图像的概率。生成对抗损失函数计算公式如下:
其中,G表示生成网络,D表示判别网络,E表示期望,x表示真实图像数据,z表示生成网络的输入数据,G(z)表示生成图像数据。通过对上述表达式执行最大最小化,即可完成对判别网络D和生成网络G参数的优化。需要指出的是D(·)表示判定括号中内容为真实图像数据的概率。
所述步骤3,具体操作如下:
在生成网络预训练模型的基础上,增加语义信息提取模块,使微调后的生成网络预训练模型更好的适应船牌文字数据集的训练。在预训练好的生成网络模型的特征提取部分,加入双向LSTM模块,对船牌文字进行语义特征的提取,从而指导网络更好的生成超分辨的船牌文字信息。LSTM模块的计算公式如下:
其中,xt表示t时刻的输入数据,w为权重,ht表示t时刻的隐含状态,h′t表示t时刻的候选隐含状态,rt表示t时刻的遗忘门,zt表示t时刻的更新门,σ表示sigmoid激活函数,tanh也是一种激活函数。
所述步骤4,具体操作如下:
在利用船牌文字数据集进行对抗学习训练时,由于文字区域具有一定的锐度,因此为了更好的指导网络生成超分辨的船牌文字,本方法增加文字梯度损失函数以加强对生成网络的监督。
梯度损失函数是一种专门为了锐化图像边缘而设计的损失,本方法采用通过图像像素平移后相减计算梯度。在RGB三通道彩色图像的基础上,分别将图像向左、向右平移一个像素,然后将得到的两幅平移图像相减便可得到原彩色图像在水平方向上的梯度。同样的,分别将图像向上、向下平移一个像素,然后将得到的两幅平移图像相减便可得到原彩色图像在垂直方向上的梯度。然后按照如下公式进行计算,便可得到最终的彩色图像梯度损失:
其中,r、l、t、b分别表示向右、向左、向上、向下平移后的图像。
通过对图像梯度的计算,可以在船牌文字数据集的训练过程中,提升船牌文字区域的锐化程度,从而提升船牌文字区域的清晰度。从而,整个网络的损失函数为:
Ltotal=λgra·Lgra+λ1·L1+λper·Lper+λadv·Ladv (18)
其中,Lgra、L1、Lper和Ladv分别表示梯度损失、L1损失、感知损失和对抗损失,λgra、λ1、λper和λadv分别表示对应损失函数的权重。
最终,通过在预训练模型的基础上更新网络结构和损失函数,进而对船牌文字数据集进行训练,使得网络能够完成对船牌文字图像的超分辨提升。
本发明的效益如下:
本发明提出了一种高性能的、专门针对于船牌文字图像的超分辨率重建方法,通过本方法,能够明显提升船牌文字的清晰程度,帮助人工快速的辨识船牌信息,方便海河航运的交通管制,提高水路航运的安全性。由于现存的船牌数据集少,而且目前的构建数据集方法并不能使船牌超分辨网络的训练达到良好效果,因此本发明通过重新采集船牌数据,采用新颖的构建船牌数据集方法,对超分辨重建网络进行训练,使船牌超分辨重建效果得到了很大提升,对同领域的研究人员有一定的借鉴意义。同时,本方法使用大型数据集DF2K对网络进行预训练,进一步提升了船牌超分辨重建的效果。不仅如此,本方法引入的双向LSTM和梯度损失,充分利用了船牌的语义信息和文字边缘特有属性,测试推理时,能在几乎不损失推理时间的前提下,进一步提升网络超分辨重建后的船牌图像质量。
本方法充分利用深度网络对图像抽象信息的提取能力,运用对抗和残差学习的思想,有效的提高了船牌图像的分辨率。通过船牌文字图像超分辨方法获取的高清晰度文字信息,能够作为船只航行过程中的证据保留。不仅如此,船牌文字图像超分辨方法在很大程度上弥补了老旧摄像头拍摄图像不清晰的缺点,在一定程度上延长了摄像头的工作寿命,提高了资源的利用率。最后,船牌文字图像超分辨技术也能够为船牌文字图像文字识别算法提供技术支持,以提高文字清晰度的方式,帮助船牌文字识别算法提高识别精度。本发明提出的方法,无论在实现难度、预算成本还是经济效益上都有明显优势。
附图说明
图1本发明的流程示意图;
图2本发明在实际场景船牌文字上的应用效果图。
具体实施方式
下面将结合附图对本发明的技术方案进行清楚、完整地描述。显然,所描述的实施方案是本发明一部分实施方案,而不是全部的实施方案。基于本发明中的实施方案,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施方案,都属于本发明保护的范围。
有鉴于此,本发明提出了一种基于语义信息和梯度监督的船牌图像超分辨方法。其主要特点在于1)通过实地考察,采集了一批船牌数据,并且经过处理,人工合成了船牌数据集。2)通过在大型数据集上进行预训练,获取良好的网络初始化权重。3)充分利用船牌文字的独特优势,提出了在网络中融合双向LSTM模块进行文字语义信息的提取,并且通过融合梯度损失函数,指导网络生成更清晰的文字边缘。在此基础上,本方法充分利用了对抗学习和残差学习的优势,完成对真实自然场景下的船牌文字图像的分辨率提升。
如图1所示,本发明的算法流程主要包括以下步骤:
步骤1,获取HR文字图像和真实模糊的船牌文字图像。在京杭大运河岸拍摄过往货船(含船牌文字)照片,通过人工标注船牌文字的中心位置坐标,剪裁出以船牌文字为中心的固定大小区域,作为HR文字图像。然后对HR文字图像进行原生噪声提取并保存,接着对HR文字图像进行降采样和添加噪声,获得最终的LR文字图像,至此由HR-LR对构成的船牌文字数据集便处理完毕。
步骤1.1,通过拍摄,最终获得1774张分辨率为4752*3168的船舶图像作为原生数据。然后对图像中船牌文字中心点进行标注,定位船牌文字位置。考虑到数据集的清晰度和多样性,需要从1774张图像中,根据不同光照、不同距离、不同清晰程度和不同色调,选取900+张的高清船舶图像,以备裁剪高清船牌文字图像;同样的,为了充分测试模型生成超分辨图像的能力,从1774张图像中,选取261张不同模糊程度的船舶图像,以备裁剪真实低分辨船牌文字图像。
步骤1.2,对900+张的高清船舶图像进行裁剪,可以获得480*480区域的仅含船牌文字区域的图像,同时,将部分船牌文字不完整的剪裁图像剔除,可获得最终的900张HR文字图像。对261张模糊的船舶图像进行剪裁,可以获得261张分辨率为120*120的真实模糊的船牌文字图像。
步骤1.3,设置待提取噪声的patch大小并提取噪声。根据网络设置,输入的HR文字图像将会被切分为一系列分辨率为128*128子图,相应的LR文字图像将会被切分为分辨率为32*32的子图。因此,输入噪声patch设置为32*32。然后将HR文字彩色图像转为灰度图像,将32*32大小的区域按照32像素的步长遍历整个图像,获得一些列噪声patch候选区。然后计算各候选区的均值和方差,并和设置的阈值(均值最小阈值为0,方差最大阈值为20)作比较,符合要求则将噪声patch的位置信息保留。最后,通过将噪声patch的位置信息作用到彩色HR文字图像上,获取最终的噪声patch。对所有的900张HR文字图像执行上述操作,便可完成对数据集中所有噪声patch的提取。
步骤1.4,HR文字图像降采样获取初步人工模糊图像。使用MATLAB软件中的双三次插值(bicubic)方法对HR文字图像降采样,获得120*120的初步人工模糊图像。降采样公式如下:
ID=(IHR*Kbic)↓s
其中,ID表示降采样后的初步人工模糊船牌文字图像,IHR表示高分辨船牌文字图像,Kbic表示双三次插值核,s表示降采样倍数,选择s=4。
步骤1.5,依次添加高斯噪声、JPEG压缩噪声和从HR文字图像中提取到的真实噪声patch。首先给生成的初步人工模糊图像加上均值为0、标准差为8.0的高斯噪声,然后对图像进行JPEG压缩,压缩质量为60(最高质量为100)。最后,通过从噪声patch库中随机提取噪声patch,叠加到JPEG压缩后的船牌文字图像上。高斯噪声的概率密度计算方式如下:
其中x为随机变量,p(x)为概率密度,δ为标准差,μ为均值。依次添加高斯噪声、JPEG压缩噪声和真实噪声patch的公式如下:
ILR=ID+ng+njpeg+ni,i∈{1,2,...,m}
其中,ILR表示LR文字图像,ID表示降采样后的图像,ng表示高斯噪声,njpeg表示JPEG压缩噪声,ni表示从噪声patch池中随机提取的一个噪声patch,m表示噪声patch池中的噪声patch的数量。通过上述操作,便得到了最终的LR文字图像,完成了HR-LR船牌文字数据集的构建。
步骤2,针对于船牌文字,构建输入数据形状为(N,C,H,W)的特征提取网络和超分辨重建网络,形成生成网络。然后通过DF2K数据集进行对抗学习预训练。在判别器的指导下,输入生成图像和真实图像,通过对抗学习对生成网络模型参数进行预训练,最终得到一个参数初始化良好的生成网络预训练模型。这部分主要包括判别网络的搭建和损失函数的构建。
步骤2.1,特征提取网络的搭建。首先通过第一层普通卷积模块进行低级特征的提取,改模块的设置为:输入通道为3,输出通道为64,卷积核大小为3*3,步长为1,填充为1。然后进入高级特征提取的网络部分,首先通过密集连接,搭建如表1所示的密集连接单元,使得每一层的输入特征图均为之前所有层输出特征图的和;然后对每一个密集连接单元内部进行1次短残差连接;3个密集单元构成一个残差连接小组,每个残差连接小组内部再进行1次长残差连接;堆叠23个残差连接小组便构成了生成网络的高级特征提取部分。将网络第一层卷积提取到的低级特征送入网络高级特征提取部分,便完成了图像抽象特征的提取。
表格1:密集连接单元结构
连接层 | 卷积核个数 | 卷积核大小 | 步长 | 填充 |
Conv1 | 32 | 3*3 | 1 | 1 |
Conv2 | 32 | 3*3 | 1 | 1 |
Conv3 | 32 | 3*3 | 1 | 1 |
Conv4 | 32 | 3*3 | 1 | 1 |
Conv5 | 64 | 3*3 | 1 | 1 |
LReLU | / | / | / | / |
步骤2.2,超分辨重建网络。首先构建2倍放大因子的上采样单元,通过使用最近邻差值的方式,对特征图进行2倍的放大,然后对放大2倍的特征图执行卷积操作,卷积的设置如下:输入通道为3,输出通道为64,卷积核大小为3*3,步长为1,填充为1。最终进行relu激活即可得到2倍上采样单元。通过堆叠2个2倍上采样单元,即可完成对图像的x4倍的超分辨提升。
步骤2.3,判别网络结构的搭建。首先通过64个3*3大小卷积核,在输入图像上执行步长为1填充也为1的卷积操作,提取到图像的低级特征图(64层)。然后通过9组由卷积层和2维批量标准化函数BN层构成的小单元,不断的提取输入图像中更抽象的高维特征。并且,随着网络提取图像的特征越来越抽象,特征图的层数也不断进行加深,最终获得512层的特征图,如表2所示。最后通过串联的两层全连接层,将3维特征图先后映射为1000维和100维的向量,然后在经过sigmoid激活层,便获得了该输入图像是真实图像的概率。本步骤获得的输入图像是真实图像的概率,将用于对抗损失函数计算。
表格2:对抗网络主体结构
步骤2.4,损失函数的设置。损失函数的构成如下:L1损失、感知损失和对抗损失。L1损失是深度学习中常用的损失函数,在图像超分辨领域,L1损失旨在计算两幅(HR和SR)图像对应像素的误差。为了构建出更符合人眼视觉效果的超分辨图像,本方法在损失函数中加入感知损失,并和生成对抗网络结合使用。通过对抗损失,生成器能生成更加真实的图像,而判别器也能有更强的判别图像真伪(生成的还是真实的)的能力。总的损失函数计算公式如下:
Ltotal=λ1·L1+λper·Lper+λadv·Ladv
其中,L1、Lper和Ladv分别表示L1损失、感知损失和对抗损失λ1、λper和λadv分别表示对应损失函数的权重。为了能够发挥损失函数的最大性能,本方法对各损失函数的权重设置如下值:λ1=10-2,λper=1,λadv=5*10-3。
步骤2.4.1,L1损失函数的设置。L1损失函数衡量的是两幅图像之间的空间距离,可以计算两幅图像(HR和SR)像素级别的误差,能够增强两幅图像像素级的相似度,从而指导网络更好得生成SR图像。其一般表达式如下:
L1=mean(ι(x,y))
ι(x,y)={c1,...,cN}T,ci=|xi-yi|
其中,N代表batch size,x和y代表形状相同的任意尺寸的图像,c表示两幅图像像素级别的差值,mean代表对所有元素取均值操作。
步骤2.4.2,感知损失函数的设置。感知损失函数的获取,本方法通过将HR图像和SR图像分别输入预训练好的卷积神经网络,然后提取第5层卷积后的特征图像做为感知特征。根据步骤2.4.1的L1损失计算公式,计算HR文字图像和SR图像感知特征的L1距离,即可获得感知损失。
步骤2.4.3,生成对抗网络损失函数的设置。在训练时,不同于经典计算生成对抗损失的方法,每次计算生成对抗损失函数,分别向判别器输入一个batch的真实图像和生成图像,然后做反向传播,同时优化生成网络和判别网络的参数。生成对抗损失函数计算公式如下:
其中,G表示生成网络,D表示判别网络,E表示期望,x表示真实图像数据,z表示生成网络的输入数据。通过对上述表达式执行最大最小化,即可完成对判别网络D和生成网络G参数的优化。
步骤2.5,其他训练参数的设置。训练中,初始学习率设置为10-4,并且设置学习率随训练衰减。迭代次数60000次,优化器使用Adam优化器。程序在Ubuntu发行版的Linux***中运行,GPU为NVIDIA RTX3070,训练时长约9小时。
步骤3,微调预训练网络,提取文字语义信息,使得网络适应船牌文字数据集的训练。为对船牌文字图像进行超分辨提升,本方法将在预训练好的生成网络的特征提取部分,加入双向LSTM模块,对船牌文字进行语义特征的提取,从而指导网络更好的生成超分辨的船牌文字信息。对于LSTM模块中的参数,使用xavier_normal进行初始化,batch_firsr设置为True,LSTM模块的计算公式如下:
zt=σ(Wz·[ht-1,xt])
rt=σ(Wr·[ht-1,xt])
h′t=tanh(W·[rt*ht-1,xt])
ht=(1-zt)*ht-1+zt*h′t
其中,xt表示t时刻的输入数据,w为权重,ht表示t时刻的隐含状态,h′t表示t时刻的候选隐含状态,rt表示t时刻的遗忘门,zt表示t时刻的更新门,σ表示sigmoid激活函数,tanh也是一种激活函数。
步骤4,在利用船牌文字数据集进行对抗学习训练时,由于文字区域具有一定的锐度,因此为了更好的指导网络生成超分辨的船牌文字,增加文字梯度损失函数以加强对生成网络的监督。
步骤4.1,为了锐化船牌文字边缘,本方法引入梯度损失函数。本方法采用通过图像像素平后相减计算梯度。在RGB三通道彩色图像的基础上,分别将图像向左、向右平移一个像素,然后将得到的两幅平移图像相减便可得到原彩色图像在水平方向上的梯度。同样的,分别将图像向上、向下平移一个像素,然后将得到的两幅平移图像相减便可得到原彩色图像在垂直方向上的梯度。然后按照如下公式进行计算,便可得到最终的彩色图像梯度损失:
其中,r、l、t、b分别表示向右、向左、向上、向下平移后的图像。将梯度损失融合到上述的损失函数中,得到整个损失函数:
Ltotal=λgra·Lgra+λ1·L1+λper·Lper+λadv·Ladv
其中,Lgra、L1、Lper和Ladv分别表示梯度损失、L1损失、感知损失和对抗损失,λgra、λ1、λper和λadv分别表示对应损失函数的权重。权重分别设置为λgra=10-4,λ1=10-2,λper=1,λadv=5*10-3。
步骤4.2,对船牌文字数据集使用pytorch自带的数据增强工具进行数据增强操作,通过色调、饱和度、角度3各方面对船牌文字数据集进行扩充,用增强后的船牌文字数据集在微调后的网络及损失函数上进行训练。
步骤4.3,其他训练参数的设置。训练中,初始学习率设置为10-4,并且设置学习率随训练衰减。迭代次数80000次,优化器使用Adam优化器。在Linux***中运行,GPU为NVIDIARTX3070,训练时长约7+小时。
步骤4.4,在步骤1中得到的261张真实模糊船牌文字数据上进行模型性能测试,完成船牌文字数据的最终训练,测试效果如图2所示。
此时,整个网络中的生成网络结构,便能够完成从120*120低分辨率船牌文字图像到480*480的超分辨率船牌文字图像的重建,是最终所需要的超分辨率生成网络。
以上所述仅是本发明的优选实施方式,应当指出,对于本技术领域的普通技术人员,在不脱离本发明构思的前提下,还可以做出若干改进和润饰,这些改进和润饰也应视为本发明保护范围内。
Claims (6)
1.一种基于语义信息和梯度监督的船牌图像超分辨方法,其特征在于包括以下步骤:
步骤1:收集船舶原始图像并进行预处理;
首先拍摄河道中含船牌文字的原始船舶图像,并进行人工标注船牌文字位置,裁剪得到高分辨率船牌文字区域图像,记为HR文字图像;随后对HR文字图像进行降采样操作并加入噪声,生成低分辨率船牌文字区域图像,记为LR文字图像;完成船牌文字数据集的构造;
步骤2:针对于船牌文字,构建特征提取网络和超分辨重建网络,形成生成网络模型;然后通过DF2K数据集进行对抗学习预训练,在判别器的指导下,通过对抗学习对生成网络模型参数进行预训练,得到生成网络预训练模型;
步骤3:在生成网络预训练模型基础上引入双向LSTM结构,增强对船牌文字区域的语义特征提取;采用船牌文字数据集,进行训练;
步骤4:在进行对抗学习训练时,由于船牌文字区域具有一定的锐度,因此为了更好的指导网络生成超分辨的船牌文字,增加文字梯度损失函数以加强对生成网络的监督。
2.根据权利要求1所述的一种基于语义信息和梯度监督的船牌图像超分辨方法,其特征在于所述步骤1具体操作如下:
1-1、根据不同光照、距离、清晰程度,在河道中拍摄船舶的高清图像和真实模糊图像,以增强数据集分布的多样性;然后对其中的高清图像和真实模糊图像中的船牌文字区域进行剪裁,对应得到HR文字图像和真实低分辨船牌文字图像,其中真实低分辨船牌文字图像作为测试集在训练完成后进行模型验证;
1-2、提取船牌文字图像原生噪声,根据设定的船牌文字噪声patch大小,对噪声patch的方差和均值进行限制,选取方差和均值均在设定阈值区间内的噪声patch区域并保存,遍历所有HR文字图像后得到噪声池;
船牌文字图像中噪声patch的均值、方差计算公式如下:
其中xp代表噪声patch中的第p个像素值,p代表每个噪声patch中的像素总个数,M代表噪声patch的均值,s2代表噪声patch的方差;
由于需要构建低分辨船牌文字图像,使用双三次插值的方式,对HR文字图像进行缩放,形成初步人工模糊图像;降采样公式如下:
ID=(IHR*Kbic)↓s (3)
其中,ID表示初步人工模糊图像,IHR表示HR文字图像,Kbic表示双三次插值核,s表示降采样倍数;
1-3、对初步人工模糊图像添加噪声;噪声的来源包括高斯噪声、JPEG压缩噪声和步骤1-2提取到的真实噪声;依次按照高斯噪声、JPEG压缩噪声和真实噪声的顺序,对初步人工模糊图像进行噪声添加,获取最终的LR文字图像,此时便完成对船牌文字数据集的构建;高斯噪声的概率密度计算方式如下:
其中x为随机变量,p(x)为概率密度,δ为标准差,μ为均值;依次添加高斯噪声、JPEG压缩噪声和真实噪声patch的公式如下:
ILR=ID+ng+njpeg+ni,i∈{1,2,...,m} (5)
其中,ILR表示LR文字图像,ID表示初步人工模糊图像,ng表示高斯噪声,njpeg表示JPEG压缩噪声,ni表示从噪声patch池中随机提取的一个噪声patch;m表示噪声patch池中的噪声patch的数量。
3.根据权利要求1或2所述的一种基于语义信息和梯度监督的船牌图像超分辨方法,其特征在于所述步骤2具体操作如下:
2-1、针对于船牌文字,构建输入数据形状为(N,C,H,W)的特征提取网络和超分辨重建网络,形成生成网络模型;
在特征提取部分,网络首先采用标准卷积层,对输入的DF2K数据集中的低分辨图像特征进行简单提取;然后通过一系列的卷积模块,搭建足够深的神经网络对图像的抽象特征进行提取,为保证浅层网络特征高效传播和避免梯度消失,采用残差连接和密集连接的方式,对深层网络进行优化;残差连接的计算公式如下:
xl=F(xl-1)+xl-1 (6)
其中,xl-1表示残差单元的输入特征,F(xl-1)表示学习到的加权后的残差映射,xl表示残差单元的输出特征;密集连接的计算公式如下:
xl=F([x1,x2,...,xl-1]) (7)
其中,x1,x2,...,xl-1表示密集连接单元的各个输入特征,F([x1,x2,...,xl-1])表示密集连接的映射函数,xl表示密集连接单元的输出特征;
最后,通过插值、卷积并激活的操作,对特征提取后的图像进行超分辨率重建,得到超分辨率重建图像;卷积并激活的公式如下:
其中x为输入特征,y为激活后的输出特征,w为卷积层权重,b为卷积层偏置,relu为激活函数;
2-2、通过DF2K数据集进行对抗学习预训练;
在判别网络中,输入超分辨率重建图像和DF2K数据集中的高分辨图像,通过对抗学习对生成网络模型参数进行预训练,最终得到一个参数初始化良好的生成网络预训练模型;这部分主要包括判别网络的搭建和损失函数的构建;
在搭建判别网络时,使用卷积层、BN层和线性映射层作为判别网络的主要模块;在卷积层的搭建中,使用较大的卷积核,每层卷积后再进行数据归一化操作,即加入BN层,这种操作能够稳定网络训练,增强鲁棒性;数据归一化的公式如下:
其中x为需要归一化的输入数据,y为归一化后的结果,mean(x)和var(x)分别为输入数据的均值和方差,eps为防止分母为0的量,γ和β为可训练参数;
最后,通过线性映射层,将图像特征的高维数据转换为向量,最终通过激活层将该向量映射成输入图像为真实高分辨图像的概率;线性映射层的计算公式如下:
y=x·AT+b (10)
其中x为线性映射层的输入,AT为线性映射层的权重,b为线性映射层的偏置,y为线性映射层的输出;
Sigmoid激活函数的计算公式如下:
其中f(x)为概率;
2-3、在预训练网络中,损失函数主要由L1损失、感知损失和对抗损失三部分构成;总的损失函数计算公式如下:
Ltotal=λ1·L1+λper·Lper+λadv·Ladv (12)
其中,L1、Lper和Ladv分别表示L1损失、感知损失和对抗损失,λ1、λper和λadv分别表示对应损失函数的权重。
4.根据权利要求3所述的一种基于语义信息和梯度监督的船牌图像超分辨方法,其特征在于步骤2-3所述损失函数具体操作如下:
①L1损失函数计算公式如下:
L1=mean(ι(x,y)) (13)
ι(x,y)={c1,...,cN}T,ci=|xi-yi| (14)
其中,N代表batch size,x和y代表形状相同的任意尺寸的图像,c表示两幅图像像素级别的差值,mean代表对所有元素取均值操作;
②感知损失通过将HR图像和SR图像分别输入预训练好的深度网络,然后提取中间层卷积后的特征图像做为感知特征;最终根据L1损失计算公式,计算HR图像和SR图像感知特征的L1距离,即可获得感知损失;
③对抗损失衡量的是网络判别器的对图像判别结果的误差;在训练时,每次计算生成对抗损失函数,分别向判别器输入一个batch的真实图像和生成图像,判断输入图像是真实图像的概率;生成对抗损失函数计算公式如下:
其中,G表示生成网络,D表示判别网络,E表示期望,x表示真实图像数据,z表示生成网络的输入数据,G(z)表示生成图像数据;通过对上述表达式执行最大最小化,即可完成对判别网络D和生成网络G参数的优化;需要指出的是D(·)表示判定括号中内容为真实图像数据的概率。
6.根据权利要求5所述的一种基于语义信息和梯度监督的船牌图像超分辨方法,其特征在于步骤4具体操作如下:
利用船牌文字数据集进行对抗学习训练时,由于文字区域具有一定的锐度,因此通过增加文字梯度损失函数以加强对生成网络的监督;
具体采用通过图像像素平移后相减计算梯度;在RGB三通道彩色图像的基础上,分别将图像向左、向右平移一个像素,然后将得到的两幅平移图像相减便可得到原彩色图像在水平方向上的梯度;同样的,分别将图像向上、向下平移一个像素,然后将得到的两幅平移图像相减便可得到原彩色图像在垂直方向上的梯度;然后按照如下公式进行计算,便可得到最终的彩色图像梯度损失:
其中,r、l、t、b分别表示向右、向左、向上、向下平移后的图像;
通过对图像梯度的计算,能够在船牌文字数据集的训练过程中,提升船牌文字区域的锐化程度,从而提升船牌文字区域的清晰度;从而,整个网络的损失函数为:
Ltotal=λgra·Lgra+λ1·L1+λper·Lper+λadv·Ladv (18)
其中,Lgra、L1、Lper和Ladv分别表示梯度损失、L1损失、感知损失和对抗损失,λgra、λ1、λper和λadv分别表示对应损失函数的权重;
最终,通过在预训练模型的基础上更新网络结构和损失函数,进而对船牌文字数据集进行训练,使得网络能够完成对船牌文字图像的超分辨提升。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202111038079.3A CN113935899A (zh) | 2021-09-06 | 2021-09-06 | 一种基于语义信息和梯度监督的船牌图像超分辨方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202111038079.3A CN113935899A (zh) | 2021-09-06 | 2021-09-06 | 一种基于语义信息和梯度监督的船牌图像超分辨方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN113935899A true CN113935899A (zh) | 2022-01-14 |
Family
ID=79275110
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202111038079.3A Pending CN113935899A (zh) | 2021-09-06 | 2021-09-06 | 一种基于语义信息和梯度监督的船牌图像超分辨方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113935899A (zh) |
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN114677569A (zh) * | 2022-02-17 | 2022-06-28 | 之江实验室 | 一种基于特征解耦合的文字-图像对生成方法和装置 |
CN116702876A (zh) * | 2023-04-27 | 2023-09-05 | 贵州大学 | 一种基于预处理的图像对抗防御方法 |
-
2021
- 2021-09-06 CN CN202111038079.3A patent/CN113935899A/zh active Pending
Cited By (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN114677569A (zh) * | 2022-02-17 | 2022-06-28 | 之江实验室 | 一种基于特征解耦合的文字-图像对生成方法和装置 |
CN114677569B (zh) * | 2022-02-17 | 2024-05-10 | 之江实验室 | 一种基于特征解耦合的文字-图像对生成方法和装置 |
CN116702876A (zh) * | 2023-04-27 | 2023-09-05 | 贵州大学 | 一种基于预处理的图像对抗防御方法 |
CN116702876B (zh) * | 2023-04-27 | 2024-04-12 | 贵州大学 | 一种基于预处理的图像对抗防御方法 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN109086824A (zh) | 一种基于卷积神经网络的海底底质声呐图像分类方法 | |
CN110458844A (zh) | 一种低光照场景的语义分割方法 | |
CN113935899A (zh) | 一种基于语义信息和梯度监督的船牌图像超分辨方法 | |
CN110363068B (zh) | 一种基于多尺度循环生成式对抗网络的高分辨行人图像生成方法 | |
CN112561876B (zh) | 基于图像的池塘和水库的水质检测方法及*** | |
CN111553837A (zh) | 一种基于神经风格迁移的艺术文本图像生成方法 | |
CN108564120A (zh) | 基于深度神经网络的特征点提取方法 | |
CN111986108A (zh) | 一种基于生成对抗网络的复杂海空场景图像去雾方法 | |
CN111489405B (zh) | 基于条件增强生成对抗网络的人脸草图合成*** | |
CN111242061B (zh) | 一种基于注意力机制的合成孔径雷达舰船目标检测方法 | |
CN114463492B (zh) | 一种基于深度学习的自适应通道注意力三维重建方法 | |
CN115471746A (zh) | 一种基于深度学习的船舶目标识别检测方法 | |
CN114066871B (zh) | 一种新冠肺炎病灶区域分割模型训练的方法 | |
CN109903373A (zh) | 一种基于多尺度残差网络的高质量人脸生成方法 | |
CN111612906A (zh) | 一种三维地质模型的生成方法、***及计算机存储介质 | |
CN111145348A (zh) | 自适应作战场景的可视化生成方法 | |
CN109740539A (zh) | 基于超限学习机和融合卷积网络的3d物体识别方法 | |
CN116664435A (zh) | 一种基于多尺度人脸解析图融入的人脸复原方法 | |
CN116468995A (zh) | 一种联合slic超像素和图注意力网络的声呐图像分类方法 | |
CN115713469A (zh) | 基于通道注意力和形变生成对抗网络的水下图像增强方法 | |
CN113807497B (zh) | 一种增强纹理细节的非配对图像翻译方法 | |
CN114943646A (zh) | 基于纹理导向的梯度权重损失和注意力机制超分辨方法 | |
CN113344110B (zh) | 一种基于超分辨率重建的模糊图像分类方法 | |
CN110675311A (zh) | 一种素描序约束下的素描生成的方法、装置及存储介质 | |
CN110991230A (zh) | 基于旋转候选框的任意方向遥感影像舰船检测方法及*** |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination |