舰船科学技术  2023, Vol. 45 Issue (7): 166-169    DOI: 10.3404/j.issn.1672-7649.2023.07.032   PDF    
改进神经网络的船舶红外图像边缘检测方法
刘志东1,2     
1. 四川大学,四川 成都 610065;
2. 四川城市职业学院,四川 成都 610110
摘要: 研究改进神经网络的船舶红外图像边缘检测方法,提升边缘检测抗噪声干扰能力。采用块匹配的主成分分析方法对船舶红外图像实施去噪处理后,经梯度算子将降噪后船舶红外图像转换为二值图像;以BP神经网络为基础,通过附加动量法-自适应学习速率调整BP神经网络权值,提高网络训练鲁棒性;将转换后舰船二值图像作为改进神经网络的输入,在实施网络训练后得出输出值,依据输出值和设置阈值的对比结果,获取船舶红外图像边缘点,实现船舶红外图像边缘检测。实验结果表明:该方法降噪后船舶红外图像的PSNR值全部高于40 dB,降噪效果较好;可有效提取船舶红外图像边缘特征且边缘检测结果清晰、连贯,能够达到船舶红外图像边缘检测标准。
关键词: 改进神经网络     船舶红外图像     边缘检测     主成分分析     学习速率     二值图像    
Improved neural network for ship infrared image edge detection
LIU Zhi-dong1,2     
1. Sichuan University, Chengdu 610065, China;
2. Urban Vocational College of Sichuan, Chengdu 610110, China
Abstract: The edge detection method of ship infrared image based on improved neural network is studied to improve the anti-noise ability of edge detection. After de-noising the ship infrared image using the block matching principal component analysis method, the de-noised ship infrared image is converted into binary image by gradient operator. Based on BP neural network, the weight of BP neural network is adjusted by the additional momentum method - adaptive learning rate to improve the robustness of network training. The converted ship binary image is taken as the input of the improved neural network, and the output value is obtained after the implementation of network training. According to the comparison results of the output value and the set threshold value, the edge points of the ship infrared image are obtained to achieve the edge detection of the ship infrared image. The experimental results show that the PSNR value of the ship infrared image after noise reduction is all higher than 40 dB, and the noise reduction effect is good. It can effectively extract the edge features of ship infrared image, and the edge detection results are clear, high and consistent, which can meet the ship infrared image edge detection standards.
Key words: improved neural network     infrared image of ship     edge detection     principal component analysis     learning rate     binary image    
0 引 言

红外图像采集系统具有采集图像速度快、不受天气影响等优点,被应用在船舶检测领域。通过红外图像采集系统可准确采集海上船舶红外图像,便于实施海上船舶管理。局部极值点汇聚一起形成图像边缘[1],因船舶红外图像边缘中存在关键的特征数据,因此对船舶红外图像边缘检测至关重要[2]。图像处理中通过边缘检测方法提取图像边缘中背景和目标灰度差异部分,有利于目标识别[3-4]。目前,相关学者已经开始对船舶红外图像边缘检测方法展开研究。王睿男[5]研究形态学的边缘检测方法,采用多方面元素提取船舶红外图像边缘特征,经多特征的加权融合计算后,完成船舶红外图像边缘检测。因该方法加权求解过程过于复杂,导致船舶红外图像边缘检测速度比较慢。何谦等[6]研究DexiNed改进的边缘检测方法,通过优化损失函数改进DexiNed方法,调整网络模型,对该模型实施训练后输出船舶红外图像边缘数据,实现船舶红外图像边缘检测。虽然该方法在少量的样本船舶红外图像边缘检测方面取得一定应用效果,但对于海量样本的船舶红外图像边缘检测中应用效果不太理想,边缘特征提取过慢,强红外图像边缘检测效率低,适应性低。

改进神经网络具有收敛性快、寻优能力强、学习能力强等优点,被广泛应用在图像识别领域,并取得较好应用效果。针对以上方法在船舶红外图像边缘检测中存在的局限性,研究改进神经网络的船舶红外图像边缘检测方法,提高船舶红外图像边缘检测能力。

1 船舶红外图像边缘检测方法 1.1 船舶红外图像去噪处理

因船舶红外图像在采集过程中容易受到外界因素的影响,导致采集的船舶红外图像中存在大量噪声数据,因此采用基于块匹配的主成分分析(principal component analysis,PCA)图像去噪方法对船舶红外图像实施去噪处理,为船舶红外图像边缘检测奠定基础。将船舶红外图像中需要去噪的像素点分为1个变量块作为训练样本,设置船舶红外图像中用 $ Y $ 表示矩阵,行列数量分别用 $ a $ $ b $ 描述,PCA变换求解流程为:

步骤1 求解船舶红外图像 $ Y $ 矩阵的各行均值,其求解公式如下:

$ {\mu _i} = \left( {1/a} \right)\sum\limits_{j = 1}^a {{Y_i}} \left( j \right)。$ (1)

其中:像素点分别用 $ i $ $ j $ 描述。

中心矩阵计算公式如下:

$ {\bar Y_i} = {Y_i} - {\mu _i}。$ (2)

步骤2 求解船舶红外图像 $ Y $ 矩阵的协方差矩阵,其求解公式为:

$ {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)

$ {D_Y} $ 特征值对角矩阵表达式如下:

$ \varPsi = {\rm{diag}}\left( {{\nu _1},{\nu _2}, \cdots {\nu _m}} \right),$ (4)

式中,特征值用 $ \nu $ 描述。

特征向量矩阵表达式为:

$ {\boldsymbol{\varsigma }} = \left[ {{\varsigma _1},{\varsigma _2}, \cdots ,{\varsigma _m}} \right]。$ (5)

特征向量用 $ \varsigma $ 描述,依据大小顺序对 $ \nu $ $ \varsigma $ 进行排序。

步骤3 投影矩阵的生成,投影矩阵表达式为:

$ \bar Z = {{\boldsymbol{\varsigma }}^{\rm{T}}}\bar Y 。$ (6)

PCA反变换方法:设置变换矩阵用 ${\boldsymbol{P}} = {{\boldsymbol{\varsigma}} ^{\rm{T}}}$ 描述,此时原始矩阵经过降维后得出投影矩阵。PCA含噪声矩阵 ${\boldsymbol{Y}}_a^{}$ 通过最小均方误差估计算法对其实施降噪,区分船舶红外图像中图像数据与噪声数据。用 ${\boldsymbol{Y}}_a^{\left( l \right)}$ 表示 ${\boldsymbol{Y}}_a^{}$ $ l $ 行,该算法表达式如下:

$ {\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)

式中:均值权重系数分别用 $ E[·] $ $ {\omega _l} $ 描述;无噪声估计值用 $ {\hat Y_l} $ 描述。

$ \sigma $ 表示噪声标准差,方差用var()描述,权重系数求解公式:

$ {\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)

其中:含噪声像素值用 $ s $ 描述;无噪声像素值用 $ o $ 描述。

通过式(7)可去除船舶红外图像矩阵各行的噪声后,采用PCA反变换后,可获取去噪后船舶红外图像。

1.2 改进神经网络的图像边缘检测 1.2.1 改进BP神经网络

BP神经网络算法流程如图1所示。BP神经网络算法流程为:将神经元实施初始化,计算隐含层、输出层每个节点的输出后,给定输入向量与目标输出,并计算目标值和实际输出的偏差,求解反向误差进行取值学习,对学习结果实施判断,满足终止条件,停止迭代,反之需要重新计算隐含层、输出层每个节点的输出。

图 1 BP神经网络算法流程 Fig. 1 BP neural network algorithm flow

通过附加动量法-自适应学习速率对BP神经网络实施改进,调整BP神经网络权值,提高网络训练鲁棒性。附加动量法权值调整算法主要是通过改变学习速率,增强网络性能,用 $ \Delta M $ 表示网络权值增量,附加动力因子权值调整表达式如下:

$ \Delta M\left( {k + 1} \right) = \left( {1 - nc} \right)\gamma *F\left( k \right) + nc*\Delta M\left( k \right)。$ (9)

式中:动量因子用 $ nc $ 描述;网络权值用 $ M $ 描述;训练次数用 $ k $ 描述;误差函数用 $ F $ 描述;学习速率用 $ \gamma $ 描述。

在网络训练时,需要对附加动力因子设置判断条件,才能正确使用权值调整公式,其判断条件公式如下:

$ {{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)

式中,第 $ k $ 代误差平方和用F(k)描述。

学习速率过大过小均会影响神经网络学习训练效率,因此需要自适应调节学习速率,学习速率自适应调整表达式如下:

$ \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)
1.2.2 图像边缘检测

改进神经网络的船舶红外图像边缘检测基本流程为:将降噪处理后船舶红外图像转换为二值图像后作为改进BP神经网络的输入,经过网络学习训练,完成船舶红外图像检测。输入层确定流程如下:

步骤1 输入预处理后船舶红外图像 $ \alpha $ ,用 $ \alpha \left( {i,j,g} \right),g = 1,2,3 $ 表示图像像素点 $ \left( {i,j} \right) $ 的彩色分量;

步骤2 通过彩色梯度算子求解出船舶红外图像中 $ \left( {{i_1},{j_1}} \right) $ $ \left( {{i_2},{j_2}} \right) $ 像素点的欧式距离,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)

经梯度算子处理后将船舶红外图像 $ \alpha $ 转换为灰度图像。

步骤3 采用梯度算子将灰度图像 $ {\alpha _1} $ 转换成二值图像 $ {\alpha _2} $

步骤4 输入样本为转换后二值图像,再实施下一步网络训练。

隐含层确定:依据相关资料可知,隐含层数量求解公式如下:

$ {\rho _1} = \sqrt {\beta + \chi } + \varepsilon 。$ (13)

其中:整数用 $ \varepsilon $ 描述 $ \varepsilon = \left[ {0 - 10} \right] $ ;输出和输入神经元数量分别用 $ \beta $ $ \chi $ 描述。此处构建由3层隐含层组成的神经网络。

输出层确定:神经网络输入层、输出层中分别包括9个神经元、1个神经元。当输出值比设置阈值0.1低时,表示检测点不是船舶红外图像边缘点;当输出值比设置阈值0.9高时,表示检测点为船舶红外图像边缘点。

经过上述过程,实现船舶红外图像检测。

2 实验分析

在MatLab 7.0软件中测试本文方法的应用效果,将某地区海事管理部门实际采集船舶红外图像数据库作为试验对象,该数据库中包括不同距离采集的船舶红外图像,试验数据库见表1

表 1 试验数据库 Tab.1 Test database

设置分别用 $ W $ $ B $ 表示 $ \iota * \tau $ 的噪声图像与干净图像,均方差误差(MSE)表达式如下:

$ 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)

其中,最大像素值用 $ MAX_B^{} $ 描述。当 $ PSNR > 40 $ 表示图像质量极好, $ 20 < PSNR < 40 $ 表示图像质量差。

采用基于块匹配的PCA算法对试验数据库中船舶红外图像进行降噪处理后,取试验数据的平均值,统计出本文方法对于不同距离采集船舶红外图像降噪后PSNR,结果见表2。分析可知,采用基于块匹配的PCA算法对于不同距离采集船舶红外图像实施降噪后PSNR值均有所提升,本文方法降噪后PSNR值全部高于40 dB,说明本文方法降噪效果较好,同时适用于不同噪声标准差下的船舶红外噪声图像去噪。

表 2 不同距离采集船舶红外图像降噪PSNR(dB) Tab.2 PSNR of infrared image noise reduction of ships collected at different distances(dB)

在试验数据库中随机选取1张近距离采集的船舶红外图像,采用本文方法对其红外图像实施边缘检测,边缘检测结果见图2。分析图2(a)可知,受雾天的影响采集原始船舶红外图像中存在大量噪声,同时图像清晰度较低;由图2(b)可知,采用基于块匹配的PCA算法对近距离采集原始船舶红外图像实施降噪处理,经PCA反转换有效去除原始图像中多余噪声,保留较好船舶红外图像边缘数据,经去噪后图像清晰度得到显著提高,视觉效果优良;由图2(c)可知,采用梯度算子将降噪后船舶红外图像转换为二值图像,保留完整的船舶红外图像中数据,为改进神经网络的检测奠定基础;由图2(d)可知,将二值图像作为改进神经网络输入,经过学习训练输出船舶红外图像边缘检测结果,采用本文方法可准确检测出船舶红外图像边缘,边缘检测结果的清晰度高且连贯,达到船舶红外图像边缘检测标准,有利于船舶分类管理。

图 2 本文方法的船舶红外图像边缘检测结果 Fig. 2 Edge detection results of ship infrared image based on this method
3 结 语

因船舶红外图像中含有大量噪声数据,加大了船舶红外图像边缘检测难度,本文研究基于改进神经网络的船舶红外图像边缘检测方法,提高船舶红外图像边缘检测速度,便于对船舶管理。

试验结果表明,本文方法可有效提取船舶红外图像真实边缘,其应用效果较好。

参考文献
[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.