2. 四川城市职业学院,四川 成都 610110
2. Urban Vocational College of Sichuan, Chengdu 610110, China
红外图像采集系统具有采集图像速度快、不受天气影响等优点,被应用在船舶检测领域。通过红外图像采集系统可准确采集海上船舶红外图像,便于实施海上船舶管理。局部极值点汇聚一起形成图像边缘[1],因船舶红外图像边缘中存在关键的特征数据,因此对船舶红外图像边缘检测至关重要[2]。图像处理中通过边缘检测方法提取图像边缘中背景和目标灰度差异部分,有利于目标识别[3-4]。目前,相关学者已经开始对船舶红外图像边缘检测方法展开研究。王睿男[5]研究形态学的边缘检测方法,采用多方面元素提取船舶红外图像边缘特征,经多特征的加权融合计算后,完成船舶红外图像边缘检测。因该方法加权求解过程过于复杂,导致船舶红外图像边缘检测速度比较慢。何谦等[6]研究DexiNed改进的边缘检测方法,通过优化损失函数改进DexiNed方法,调整网络模型,对该模型实施训练后输出船舶红外图像边缘数据,实现船舶红外图像边缘检测。虽然该方法在少量的样本船舶红外图像边缘检测方面取得一定应用效果,但对于海量样本的船舶红外图像边缘检测中应用效果不太理想,边缘特征提取过慢,强红外图像边缘检测效率低,适应性低。
改进神经网络具有收敛性快、寻优能力强、学习能力强等优点,被广泛应用在图像识别领域,并取得较好应用效果。针对以上方法在船舶红外图像边缘检测中存在的局限性,研究改进神经网络的船舶红外图像边缘检测方法,提高船舶红外图像边缘检测能力。
1 船舶红外图像边缘检测方法 1.1 船舶红外图像去噪处理因船舶红外图像在采集过程中容易受到外界因素的影响,导致采集的船舶红外图像中存在大量噪声数据,因此采用基于块匹配的主成分分析(principal component analysis,PCA)图像去噪方法对船舶红外图像实施去噪处理,为船舶红外图像边缘检测奠定基础。将船舶红外图像中需要去噪的像素点分为1个变量块作为训练样本,设置船舶红外图像中用
步骤1 求解船舶红外图像
$ {\mu _i} = \left( {1/a} \right)\sum\limits_{j = 1}^a {{Y_i}} \left( j \right)。$ | (1) |
其中:像素点分别用
中心矩阵计算公式如下:
$ {\bar Y_i} = {Y_i} - {\mu _i}。$ | (2) |
步骤2 求解船舶红外图像
$ {D_Y} = E\left( {\bar Y{{\bar Y}^{\rm{T}}}} \right) = \frac{1}{a}\bar Y{\bar Y^{\rm{T}}},\bar Y = {\left[ {\bar Y_1^{\rm{T}}\bar Y_2^{\rm{T}} \cdots \bar Y_m^{\rm{T}}} \right]^{\rm{T}}},$ | (3) |
$ \varPsi = {\rm{diag}}\left( {{\nu _1},{\nu _2}, \cdots {\nu _m}} \right),$ | (4) |
式中,特征值用
特征向量矩阵表达式为:
$ {\boldsymbol{\varsigma }} = \left[ {{\varsigma _1},{\varsigma _2}, \cdots ,{\varsigma _m}} \right]。$ | (5) |
特征向量用
步骤3 投影矩阵的生成,投影矩阵表达式为:
$ \bar Z = {{\boldsymbol{\varsigma }}^{\rm{T}}}\bar Y 。$ | (6) |
PCA反变换方法:设置变换矩阵用
$ {\hat Y_l} = {\omega _l}\left\{ {{\boldsymbol{Y}}_a^{\left( l \right)} - E\left[ {{\boldsymbol{Y}}_a^{\left( l \right)}} \right]} \right\} + E\left[ {{\boldsymbol{Y}}_a^{\left( l \right)}} \right]。$ | (7) |
式中:均值权重系数分别用
$ {\omega _l} = {var} \left( s \right)/{var} \left( o \right) = \frac{{{var} \left( o \right) - {\sigma ^2}}}{{E{{\left\{ {{\boldsymbol{Y}}_a^{\left( l \right)} - E\left[ {{\boldsymbol{Y}}_a^{\left( l \right)}} \right]} \right\}}^2}}} $ | (8) |
其中:含噪声像素值用
通过式(7)可去除船舶红外图像矩阵各行的噪声后,采用PCA反变换后,可获取去噪后船舶红外图像。
1.2 改进神经网络的图像边缘检测 1.2.1 改进BP神经网络BP神经网络算法流程如图1所示。BP神经网络算法流程为:将神经元实施初始化,计算隐含层、输出层每个节点的输出后,给定输入向量与目标输出,并计算目标值和实际输出的偏差,求解反向误差进行取值学习,对学习结果实施判断,满足终止条件,停止迭代,反之需要重新计算隐含层、输出层每个节点的输出。
通过附加动量法-自适应学习速率对BP神经网络实施改进,调整BP神经网络权值,提高网络训练鲁棒性。附加动量法权值调整算法主要是通过改变学习速率,增强网络性能,用
$ \Delta M\left( {k + 1} \right) = \left( {1 - nc} \right)\gamma *F\left( k \right) + nc*\Delta M\left( k \right)。$ | (9) |
式中:动量因子用
在网络训练时,需要对附加动力因子设置判断条件,才能正确使用权值调整公式,其判断条件公式如下:
$ {{nc = }}\left\{ {\begin{array}{*{20}{c}} 0,& F(k){>}1.04F(k - 1),\\ 0.95,&F(k){<}F(k - 1),\\ ne,& {{\rm{others}}}。\\ \end{array}} \right.$ | (10) |
式中,第
学习速率过大过小均会影响神经网络学习训练效率,因此需要自适应调节学习速率,学习速率自适应调整表达式如下:
$ \gamma \left( k \right) = \left\{ {\begin{array}{*{20}{c}} {1.05\gamma \left( k \right),} & {F{\rm{(}}k{\rm{) < }}F{\rm{(}}k{\rm{ - 1)}}},\\ {0.75\gamma \left( k \right),} & {F{\rm{(}}k{\rm{) > 1}}{\rm{.05}}F{\rm{(}}k{\rm{ - 1)}}},\\ {\gamma \left( {k - 1} \right),} & {{\rm{others}}}。\\ \end{array}} \right. $ | (11) |
改进神经网络的船舶红外图像边缘检测基本流程为:将降噪处理后船舶红外图像转换为二值图像后作为改进BP神经网络的输入,经过网络学习训练,完成船舶红外图像检测。输入层确定流程如下:
步骤1 输入预处理后船舶红外图像
步骤2 通过彩色梯度算子求解出船舶红外图像中
$ D\left( {{i_1},{j_1},{i_2},j_2^{}} \right) = \sqrt {{{\left[ {\alpha \left( {{i_1},{j_1},1} \right) - \alpha \left( {{i_1},{j_1},1} \right)} \right]}^2} + {{\left[ {\alpha \left( {{i_1},{j_1},2} \right) - \alpha \left( {{i_1},{j_1},2} \right)} \right]}^2} + {{\left[ {\alpha \left( {{i_1},{j_1},3} \right) - \alpha \left( {{i_1},{j_1},3} \right)} \right]}^2}}。$ | (12) |
经梯度算子处理后将船舶红外图像
步骤3 采用梯度算子将灰度图像
步骤4 输入样本为转换后二值图像,再实施下一步网络训练。
隐含层确定:依据相关资料可知,隐含层数量求解公式如下:
$ {\rho _1} = \sqrt {\beta + \chi } + \varepsilon 。$ | (13) |
其中:整数用
输出层确定:神经网络输入层、输出层中分别包括9个神经元、1个神经元。当输出值比设置阈值0.1低时,表示检测点不是船舶红外图像边缘点;当输出值比设置阈值0.9高时,表示检测点为船舶红外图像边缘点。
经过上述过程,实现船舶红外图像检测。
2 实验分析在MatLab 7.0软件中测试本文方法的应用效果,将某地区海事管理部门实际采集船舶红外图像数据库作为试验对象,该数据库中包括不同距离采集的船舶红外图像,试验数据库见表1。
设置分别用
$ MSE = \frac{1}{{\iota * \tau }}\sum\limits_{\iota = 0}^{\iota - 1} {} \sum\limits_{\tau = 0}^{\tau - 1} {} {\left[ {B\left( {\iota ,\tau } \right) - W\left( {\iota ,\tau } \right)} \right]^2},$ | (14) |
峰值信噪比(PSNR)作为评价指标,其计算公式如下:
$ PSNR = 10*{\rm{lo{g}}_{10}}\left( {\frac{{MAX_B^2}}{{MSE}}} \right) = 20*{\rm{lo{g}}_{10}}\left( {MAX_B^{}/\sqrt {MSE} } \right)。$ | (15) |
其中,最大像素值用
采用基于块匹配的PCA算法对试验数据库中船舶红外图像进行降噪处理后,取试验数据的平均值,统计出本文方法对于不同距离采集船舶红外图像降噪后PSNR,结果见表2。分析可知,采用基于块匹配的PCA算法对于不同距离采集船舶红外图像实施降噪后PSNR值均有所提升,本文方法降噪后PSNR值全部高于40 dB,说明本文方法降噪效果较好,同时适用于不同噪声标准差下的船舶红外噪声图像去噪。
在试验数据库中随机选取1张近距离采集的船舶红外图像,采用本文方法对其红外图像实施边缘检测,边缘检测结果见图2。分析图2(a)可知,受雾天的影响采集原始船舶红外图像中存在大量噪声,同时图像清晰度较低;由图2(b)可知,采用基于块匹配的PCA算法对近距离采集原始船舶红外图像实施降噪处理,经PCA反转换有效去除原始图像中多余噪声,保留较好船舶红外图像边缘数据,经去噪后图像清晰度得到显著提高,视觉效果优良;由图2(c)可知,采用梯度算子将降噪后船舶红外图像转换为二值图像,保留完整的船舶红外图像中数据,为改进神经网络的检测奠定基础;由图2(d)可知,将二值图像作为改进神经网络输入,经过学习训练输出船舶红外图像边缘检测结果,采用本文方法可准确检测出船舶红外图像边缘,边缘检测结果的清晰度高且连贯,达到船舶红外图像边缘检测标准,有利于船舶分类管理。
因船舶红外图像中含有大量噪声数据,加大了船舶红外图像边缘检测难度,本文研究基于改进神经网络的船舶红外图像边缘检测方法,提高船舶红外图像边缘检测速度,便于对船舶管理。
试验结果表明,本文方法可有效提取船舶红外图像真实边缘,其应用效果较好。
[1] |
李晓峰, 李东, 王妍玮. 基于全卷积神经网络的医疗图像弱边缘检测算法[J]. 哈尔滨理工大学报, 2021, 26(3): 65-73. LI Xiao-feng, LI Dong, WANG Yan-wei. Weak edge detection algorithm for medical images based on full convolution neural network[J]. Journal of Harbin University of Science and Technology, 2021, 26(3): 65-73. |
[2] |
李翠锦, 瞿中. 基于卷积神经网络的跨层融合边缘检测算法[J]. 计算机应用研究, 2021, 38(7): 2183-2187. LI Cui-jin, QU Zhong. Cross fusion edge detection algorithm based on CNN[J]. Application Research of Computers, 2021, 38(7): 2183-2187. |
[3] |
岳欣华, 邓彩霞, 张兆茹. BP神经网络与形态学融合的边缘检测算法[J]. 哈尔滨理工大学学报, 2021, 26(5): 83-90. YUE Xin-hua, DENG Cai-xia, ZHANG Zhao-ru. BP Neural Network fuse with morphology edge detection method[J]. Journal of Harbin University of Science and Technology, 2021, 26(5): 83-90. |
[4] |
张宪红. 一种基于反应扩散方程的彩色图像边缘检测方法[J]. 吉林大学学报(理学版), 2021, 59(6): 1481-1490. ZHANG Xian-hong. A color image edge detection method based on reaction diffusion equation[J]. Journal of Jilin University(Science Edition), 2021, 59(6): 1481-1490. |
[5] |
王睿男, 武穆清, 陈铁英, 等. 基于形态学的红外图像边缘检测[J]. 北京邮电大学学报, 2021, 44(1): 66-71. WANG Rui-nan, WU Mu-qing, CHEN Tie-ying, et al. Edge detection of infrared image based on morphology[J]. Journal of Beijing University of Posts and Telecommunications, 2021, 44(1): 66-71. |
[6] |
何谦, 刘伯运. 一种基于DexiNed改进的红外图像边缘检测算法[J]. 红外技术, 2021, 43(9): 876-884. HE Qian, LIU Bo-yun. Improved infrared image edge detection algorithm based on DexiNed[J]. Infrared Technology, 2021, 43(9): 876-884. |