收藏 分销(赏)

内河航运密集小目标船舶的图像检测方法.pdf

上传人:自信****多点 文档编号:2613337 上传时间:2024-06-03 格式:PDF 页数:9 大小:3.38MB
下载 相关 举报
内河航运密集小目标船舶的图像检测方法.pdf_第1页
第1页 / 共9页
内河航运密集小目标船舶的图像检测方法.pdf_第2页
第2页 / 共9页
内河航运密集小目标船舶的图像检测方法.pdf_第3页
第3页 / 共9页
亲,该文档总共9页,到这儿已超出免费预览范围,如果喜欢就下载吧!
资源描述

1、第 32 卷 第 1 期厦门理工学院学报Journal of Xiamen University of TechnologyVol.32 No.12024 年 2 月Feb.2024内河航运密集小目标船舶的图像检测方法吴志华1,钟铭恩1*,邓智颖1,吴航星2,谭佳威1,周美华3(1.厦门理工学院机械与汽车工程学院,福建 厦门 361024;2.厦门理工学院光电与通信工程学院,福建 厦门 361024;3.皖南医学院医学信息学院,安徽 芜湖 241002)摘 要 针对小目标船舶在密集性场景中检测的漏检率和误检率较高的问题,基于深度卷积神经网络,提出一种船舶检测方法。该方法选用YOLOv7模型作为

2、算法参考框架进行改进,设计自适应锚框匹配算法,利用CIoU距离度量方式重构K-means聚类算法,以更好地适应船舶数据集中物体的尺寸和比例分布;增加一个针对小尺度目标的细粒度检测头,并使用分散注意力机制对网络模型结构进行重新设计;采用负片图像增强技术来扩充数据样本以获得更多训练样本。实验结果表明:与YOLOv7原模型相比,本文算法在内河航运船舶检测任务中的查准率和查全率分别达到92.8%和88.9%,分别提高10.9%和23.6%;mAP50达到92.6%,提高23.4%;FPS指标下降11.4;模型大小为47.1 MB,在PC端上单张图片耗时需要32.26 ms,实现了小目标船舶的高效检测。

3、关键词 内河航运;船舶检测;YOLOv7算法模型;图像增强中图分类号 TP399 文献标志码 A 文章编号 1673-4432(2024)01-0058-09Image Detection Method for Dense Small Target Ships in Inland Waterway NavigationWU Zhihua1,ZHONG Ming en1*,DENG Zhiying1,WU Hangxing2,TAN Jiawei1,ZHOU Meihua3(1.School of Mechanical and Automotive Engineering,Xiamen Univ

4、ersity of Technology,Xiamen 361024,China;2.School of Opto-Electronic and Communication Engineering,Xiamen University of Technology,Xiamen 361024,China;3.School of Medical Informatics,Wannan Medical College,Wuhu 241002,China)Abstract:A ship detection method based on deep convolutional neural network

5、is proposed to solve the problem of high miss rate and high false alarm rate of small target ships in dense scenes.The YOLOv7 model is selected as the reference frame of the algorithm to improve the method.Firstly,we design an adaptive anchor boxes matching algorithm that reconfigures the K-means cl

6、ustering algorithm using the CIoU distance metric to better match the size and scale distribution of objects in our ship dataset.Secondly,we add a fine-grained detection head for small-scale targets and redesign the network model structure using the shuffle attention(SA)doi:10.19697/ki.1673-4432.202

7、401008收稿日期:20221213 修回日期:20230301基金项目:福建省自然科学基金项目(2019J01859)通信作者:钟铭恩,男,教授,博士,研究方向为机器视觉与人工智能、智能汽车等,E-mail:。引文格式:吴志华,钟铭恩,邓智颖,等.内河航运密集小目标船舶的图像检测方法 J.厦门理工学院学报,2024,32(1):58-66.Citation:WU Z H,ZHONG M E,DENG Z Y,et al.Image detection method for dense small target ships in inland waterway navigationJ.Jou

8、rnal of Xiamen University of Technology,2024,32(1):58-66.(in Chinese)第 1 期吴志华,等:内河航运密集小目标船舶的图像检测方法mechanism.Lastly,we utilize a negative image enhancement technique to expand the data samples and obtain more training examples.Experimental results show that the algorithm achieves an accuracy and reca

9、ll of 92.8%and 88.9%,respectively,for the inland waterway shipping ship detection task,representing an improvement of 10.9%and 23.6%,respectively.The mAP50 value also improves by 23.4%,reaching 92.6%,while the FPS index decreases by 11.4.Our model size is 47.1 MB,and the time required for a single i

10、mage on the PC side is 32.26 ms,achieving efficient detection of small target ships.Key words:inland waterway navigation;ship detection;YOLOv7 algorithm model;image enhancement近年来,河道运输业呈现快速增长趋势,在枢纽河段采取必要的监管策略,有助于提前预防交通事故的发生。船舶出行具有不定时、不可预见的特点。人工值守查看视频工作量大、易视疲劳,容易引起漏检和误检。因此,需要研究出智能化的船舶检测方法。目前常用的检测方法主要

11、是在固定高度的地方安装智能监控摄像头,通过网联工控机实时地对视频进行检测并发出预警1。由于设备安装点较高,这会导致视频中出现的船舶个体都为小目标,检测任务易受到附近噪声的干扰。早期利用机器视觉的船舶检测方法主要分为区域选择、特征提取和分类3个步骤2。区域划分后,通过针对性场景设计的图像算子来提取船舶的边缘、纹理、颜色等特征,进一步实现船舶分类和定位,为后续的异常事件检测提供数据支撑。不足的是,这些特征对船舶的表征能力不强,例如,常用的方向梯度直方图算子(histogram of oriented gradient,HOG)3和尺度不变特征变换(scale-invariant feature t

12、ransform,SIFT)4。河段环境的不同,其提取特征的方法也不同,容易导致算法模型难以迁移使用,实时性较低。随着智能光电监控技术的发展,利用深度学习技术实现船舶的检测已经成为研究热点。例如,薛远亮等5对SSD模型引入了融合通道和空间注意力等策略,用于船舶的实时检测;陈从平等6设计了具有后处理算法的Faster RCNN算法模型,有效降低了船舶检测的漏检率;周薇娜等7提出了一种重新设计的YOLOv4多尺度船舶实时检测方法,通过构建的双向细粒度特征金字塔,在包含多种尺度的船舶图像上,验证了算法的超高性能;李佳东等8提出一种基于YOLOv5s的船舶检测算法,通过引入空间金字塔模块,解决了船舶检

13、测的精度问题。这些改进方案的检测精度虽然达到了较优水平,但主要是针对海面近距离拍摄的情况进行设计,对于内河航运中小目标的船舶检测不再适用。为了解决该问题,有研究者们提出了其他的检测方法。例如,周慧等9提出了一种基于特征金字塔网络(feature pyramid network,FPN)模型,解决了合成孔径雷达(synthetic-aperture radar,SAR)图像中小尺度船舶难检测的问题;邓睿哲等10设计了一种用于遥感影像船舶检测的FPN建模方法,克服了网络难以应对具有随机朝向、形态多变的小尺度船舶的情况。然而,这些方法主要是针对遥感影像数据进行算法开发,难以应用在监控图像上。实际上,

14、内河河道普遍存在早晚集中出航的实况,易面临其他目标物的遮挡和周围的尾流等噪声干扰,这都增加了检测难度。因此,增强上下文语义信息,突出船舶的显著特征,对于提高小尺度船舶的检测性能至关重要,合理选择一种高精度的检测模型作为参考并在此基础上重新设计是解决这些问题的有效路径。YOLOv7算法模型11具有高精度、低耗时的性能优势,但目前学者对其研究并不多。为此,本文以该模型为参考,提出了一种小目标船舶检测方法,该方法首先通过负片图像增强算法来扩充少样本船舶标签数量,然后增加一个1010大小的细粒度检测头,检测头的锚框参数进行CIoU度量自适应优化。同时,在模型Backbone上引入分散注意力(shuff

15、le attention,SA)机制结构,完成船舶的目标检测。1YOLOv7算法模型YOLOv7是基于锚框的检测模型,它的参数量较少,占用内存空间小,适合在边缘设备上部署。该网络模型主要分为3部分:1)骨干网络(Backbone)用来提取不同输出尺度的特征;2)特征融合网络(Neck)主要负责特征的融合;3)检测头(Head)负责结果的预测输出。59厦门理工学院学报2024 年图1是YOLOv7的多尺度检测结构图,可见,模型有3个检测头。该模型每个检测头层分配3个先验框,用来回归不同尺度大小的船舶。图片输入经骨干网络和特征融合网络部分后,在3个不同尺度的检测层上输出目标的位置信息和类别信息。每

16、一个检测层输出形状为(na,fw,fh,(nc+5)的张量,其中:na代表该层用于回归的锚框数量;fw、fh分别代表输出特征图的宽、高;nc+5代表类别和位置信息,nc为类别数量,数值5表示预测框的中心点坐标、宽高尺寸和置信度的信息。特征图的每一个位置都有na个锚框来回归目标。网络输出的位置信息是相对于预先设定的锚框偏移量,需要通过锚框将预测值输出解码成检测框。2算法改进2.1结合先验框和模型尺度的结构调整通过多尺度融合检测的思想,使用K-means聚类算法对船舶数据集进行聚类。在K-means算法中,在一个固定的簇数K下,通过最小化总体平方和来求解最佳质心,然后基于质心实现锚框聚类,计算公式

17、为Cscore=1-measurement。(1)式(1)中:Cscore是质心与所有目标框的平均得分;measurement是距离度量方法。本文采用CIoU距离度量方式,具体为CIoU=IoU(b,bgt)-(l2(b,bgt)d2+v)。(2)式(2)中:bgt是真实目标框;b是聚类的质心;v是衡量长宽比的相似性因子,v=42(arctanwGThGT-arctanwcentroidhcentroid)2;是权重系数,=v(1-IoU(b,bgt)+v+),为无穷小数;l2(b,bgt)是质心与目标框中心点的欧氏距离;d为锚框与目标框的最小闭包矩形对角线距离。因此,采用CIoU距离度量方式

18、的K-Means聚类计算公式为CscoreCIoU=1-CIoU=1-IoU(b,bgt)+l2(b,bgt)d2+v。(3)式(3)中:CscoreCIoU是质心与所有目标框的平均CIoU得分。由图2中 K与平均CIoU关系曲线变化可知,真实框的平均CIoU与质心数量K呈正相关,但同时模型复杂度也会随之增加,并带来更多的负训练样本。当K大于12时,平均CIoU的提升幅度逐渐平缓。因此,本文选择的先验框个数为12,其适应 Feature Map 2020 2倍上采样 Feature Map 4040 2倍上采样 Feature Map 8080 +4倍上采样 检测头 1 检测头 2 检测头 3

19、 640640 SPPCSPC 结构 4040 80802020 8倍下 采样 图1YOLOv7的多尺度检测结构图Fig.1Multi-scale detection structure of YOLOv760第 1 期吴志华,等:内河航运密集小目标船舶的图像检测方法于4个检测尺度,以在保证检测精度的情况下使算法的检测速度达到更快。本文对 YOLO 增加一个 1010 的细粒度检测头,以帮助模型融合更多小目标深层特征语义信息,实现更高的定位精度,提升检测效率。当输入为 640640的图像尺度训练时,改进的多尺度融合网络分别通过8倍、16倍、32倍、64倍下采样的特征融合层,对应选择 8080、

20、4040、2020、1010共4种特征图分辨率进行预测。8080对应的检测头为尺度1,在K-means聚类算法中采用CIoU距离度量方式对船舶数据集聚类出的先验框结果分别为(5,3)、(6,4)、(8,4);4040为尺度2,它不仅拥有自身的特征信息,还能融合尺度1的特征经过2倍上采样的信息,其聚类结果为(8,5)、(12,5)、(10,7);同理,2020为尺度3,在自身特征信息的基础上能融合尺度1和尺度2上采样后的特征信息,其聚类结果为(12,9)、(19,7)、(15,11);增加的尺度4包含的是1010特征图与尺度1、尺度2、尺度3上采样融合后的特征信息,其对应的先验框聚类结果为(21

21、,12)、(21,17)、(58,33)。2.2优化特征融合结构注意力机制有助于神经网络准确地关注局部重要特征区域,已经成为提高深层神经网络性能的一个研究方向。常用的注意力机制主要有空间注意力和通道注意力2种,分别旨在捕捉像素级的成对关系和通道依赖性12-13。通常,将两者融合的实现方式更佳的特征提取性能,但容易增加计算负担。本文引入一种更高效的且不带来额外计算开销的SA机制14。采用组卷积的思想,使通道和空间注意力协同作用。该注意力机制的原理如图3所示。1)分组操作。获得g组不同特征图,目的是降低模型的计算量,加快网络的运行速度。分组操作之后的特征层宽度和高度不变,通道数变为cg,每个组内部

22、使用SA单元进行处理。该单元内部包含通道注意力机制和空间注意力机制,前者具体实现类似挤压和激励操作(squeeze-and-excitation,SE),如A1A2,后者使用组规范化操作实现特征提取(group norm,GN),如B1B2。2)拼接操作。将第一步经SA单元处理后得到的特征图通过Concat的方式进行拼接,实现组内部的信息融合与交互。3)重排操作。使用分散操作对组进行重排,使不同组之间进行信息流通,以输出带有注意力机制的特征层。wch Shuffle 分组 汇总 /c g/2cg组规范化(GN)分解 融合 gpf(.)c f(.)c f(.)(.)输入特征图 输出特征图 cat

23、 A1 A2 B2 B1 挤压和激励操作图3分散注意力机制的原理图Fig.3Principle of SACIoU0.80.70.60.50.40.30.20.102.5 5.0 7.5 10.0 12.5 15.0 17.5 20.0 22.5 K图2K与CIoU的关系曲线Fig.2Curve of relationship between K and CIoU61厦门理工学院学报2024 年为了验证以上方法,本文设计了一个实时船舶检测模型,它包括骨干网络、特征融合网络、检测头和注意力机制模块等,模型结构如图4所示。该模型以YOLOv7算法为参考,骨干网络在第十一层新引入一个分支,和负责输出

24、尺度为8080的检测头进行通道融合。特征融合网络主要思路为增加一层上采样的操作,为多引入1010大小的检测头提供基础条件。注意力机制模块引入的位置在MP2模块和Concat中间,本文一共添加3个该模块,实现4个检测头的特征传输。2.3船舶图像的数据增强方法本文通过固定角度的光电监控摄像头,在国内某个内河航道进行实时图像采集,得到731张不同的船舶出航场景图像,图像分辨率大小为1 920 px1 080 px。图像中船舶类别共包含私人小船、货艇和巡逻船3类,如图5所示。由图5可见,船舶目标小,且分布密集。因此,标注这种细小的目标耗费人力,容易造成标注错误。而深度学习模型的性能易受到训练标签的质量

25、和数目的影响。针对小目标学习特征不足的难题,现有研究多采用普通的图像增强技术。依据这种做法提供的小目标特征有限而无法满足本场景的需要。为此,本文提出了一种负片数据增强方法,利用图像边缘信息易突显性优势,结合负片效应和图像锐化机制,扩建出更多有效的样本。具体过程如下。输入:内河航运密集小目标船舶数据集合A=a1,a2,an。输出:原始数据集像素信息集合P=(w1,h1),(w2,h2),(wi,hi),i 1,2,m,其中:(wi,hi)为样本的宽度和高度;m为样本数量。负片数据集合为S1,高斯模糊处理后的数据集合为S2,USM锐化处理之后的目标数据集合为S3。步骤1:P ,S0 ,S1 ,S2

26、 ,S3 。当集合A非空时,重复执行以下步骤。步骤2:获取A中样本的像素信息P,再将需处理样本进行灰度图转换得到S0。步骤3:由(wi,hi)逐一遍历数据集灰度图中的像素点,之后采用式(4)MP1MP1MP1SPPCSPCELAN-HConvcatCBSBNCBSConvSiLUCBScatELAN-HMP2CBS CBS CBS CBSELAN-HUPUPMP2ELAN-HcatREPConvConvUPCBSUpsampleCBSCBSCBSMPMaxpoolSPPCSPCCBSCBSCBSCBSMaxpoolMaxpoolMaxpool1.Backbone112437502.Neck75

27、126(3,1)(3,2)(1,1)=cc/2c/2cMP1:C -C MP2:C -2Ccc/2c/2=ELAN-HMP2ConvUPcatELAN-HSASASA100113CBS=尺度4尺度3尺度2尺度13.HeadREPREPREPcatELANELANELANELANcatcatCBSCBScatcatCBSCBSCBS8763CBSCBSCBSCBSCBSCBSCBScatELAN-HELAN=c/22c图4船舶检测模型结构图Fig.4Structure of ship detection model图5船舶监控图像举例Fig.5Example of ship surveillan

28、ce image62第 1 期吴志华,等:内河航运密集小目标船舶的图像检测方法G_back(wi,hi)=255-G_forward(wi,hi)(4)对集合P中每个像素点进行灰度值反转,得出集合S1。式(4)中:G_forward(wi,hi)为集合S0中在(wi,hi)像素点反转前的灰度值;G_back(wi,hi)为集合S1中在(wi,hi)像素点反转后的灰度值。步骤4:在集合S1中使用高斯模糊处理,得到集合S2。步骤5:在集合S2中使用USM锐化增强图像算法,得到集合S3,具体公式为S3(n)=S1(n)+S2(n)。(5)式(5)中:n为三者集合对应的样本键值;和为图像增强效果的缩放

29、系数。图6为负片数据增强后的样本举例,与原图对比可见,通过负片图像增强方法的样本纹理信息较为突出。同时,本文对训练中的输入标签数据再进行Mosaic数据增强。该方法利用裁剪、融合、缩放、拼接、掩摸、模糊等图像操作,将输入图像进行处理,结果见图7。由图7可见,原本较大的船舶目标再缩放较大的系数,为模型提供更充足的小目标船舶样本标签去学习。3实验结果及分析本实验首先在 Ubuntu22.04 操作系统下进行。处理器为 Intel(R)Core(TM)i7-10700K CPU 3.80GH 3.79 GHz,内存为 32 GB,显卡型号为 NVIDIA GeForce RTX 3080。编译语言为

30、 Python3.8,深度学习框架为PyTorch1.12.1,图像加速器版本为Cuda11.0和Cudnn8。基准模型选择YOLOv7普通版本。模型训练时,输入图片大小设置为640 px640 px,默认训练轮数设置为300,每个轮次输入图像训练的批量大小设置为32,同时模型训练中采用Adam方法作为优化器。然后,将该YOLOv7改进模型部署在个人PC端设备进行测试,该设备的处理器为AMD Ryzen 7 5800H3.20 GHz,显卡型号为NVIDIA GeForce RTX 3050Ti。3.1算法评估本文根据平均精度(average precision,AP)、平均精度均值(mean

31、 average precision,mAP)、每秒检测帧数(frames per second,FPS)3 项指标对算法进行评估,进而比较及验证算法性能。AP 为 P-R(precision-recall)曲线的积分面积,其计算公式为AP=01P(R)dR。(6)式(6)中:查准率(precision,P)、查全率(recall,R)分别定义为 P=TP(TP+FP)、R=TP(TP+FN),其中:TP为真阳性样本数目;FP为假阳性样本数目;FN为假阴性样本数目。mAP为各个类别的AP均值(IoU阈值取0.5),计算公式为图6负片数据增强样本举例Fig.6Example of negativ

32、e data enhancement图7Mosaic数据增强结果Fig.7Results of Mosaic data enhancement63厦门理工学院学报2024 年mAP=1Ni=1NAP()k。(7)式(7)中:N代表待检测的船舶总类别数目,本文N取值为3。3.2自适应锚框匹配实验表1为两种不同的距离度量方法对YOLOv7算法模型检测头锚框参数优化的客观评估数据。由表 1 可见,采用CIoU距离度量方法优化锚框,其检测的查准率、查全率和全类别平均精度均 值 分 别 达 到 81.9%、65.3%和69.2%,与欧式距离方法相比,分别提高了1.6%、1.3%和1.1%。表1的数据说明

33、,在K-means聚类算法中采用本文的CIoU度量方式生成的先验框能够更好地匹配物体的尺寸和比例分布,检测效果上比欧式距离方法更优。图8是在K-means算法中采用CIoU度量方式的聚类结果。由图8可知,大部分船舶目标在图像中宽度和高度范围都在50 px以下,而该图像总分辨率为较大的1 920 px1 080 px,在这种大分辨率图像中,细小尺度占比的船舶的检测是有难度的。由图8对比可见,相比于原本3个检测头分配到的9个锚框数量分布,增加一个1010大小的检测头后,包含12个锚框数量的分布更加均衡,说明增加细粒度检测头能更有效地辅助目标检测算法定位和识别细小尺度的船舶。3.3消融实验为了量化分

34、析本文提出的各个改进环节对船舶检测算法的性能影响,在相同实验条件下开展了消融实验,结果见表2,其中打勾符号表示实验有添加的改进方法。由表2可见:通过负片图像增强方法训练得到的模型的查准率、查全率和全类别平均精度均值分别达到83.7%、67.4%和71.4%,与原始模型相比,分别提升了 1.8%、2.1%和2.2%。这表明设计的负片图像增强算法可以有效扩充目标的标注标签,促进船舶的边缘特征得到学习以提升检测精度。从表 2 还可知,增加一个细粒度检测头可以输出具备更广感受野的特征图,更准确地定位物体,从而提高检测的精度,因此使得检测算法的查准率、查全率和全类别平均精度均值分别提升了 高度/px高度

35、/px宽度/px宽度/px2502001501005000 100 200 300 400 500 600 7002502001501005000 100 200 300 400 500 600 700 (a)锚框数量为9 (b)锚框数量为12图8利用CIoU距离度量的K-Means算法聚类结果Fig.8Clustering results using K-Means algorithm of CIoU distance measurement表1不同距离度量方式的客观评估数据Table 1Objective evaluation data for different distance mea

36、sures单位:%K-means欧式距离度量CIoU的距离度量P80.381.9R64.065.3mAP5068.169.2表2消融实验结果Table 2Results of ablation experiments单位:%所用方法YOLOv7(K-means)改进1改进2本文方法图像增强HeadSAP81.983.790.992.8R65.367.481.988.9mAP5069.271.487.092.6说明:打勾符号表示实验有添加的改进方法64第 1 期吴志华,等:内河航运密集小目标船舶的图像检测方法7.2%、14.5%和15.6%。此外,引入SA注意力机制后,模型的查准率、查全率和全类

37、别平均精度均值再次分别提升1.9%、7.0%和5.6%,分别达到92.8%、88.9%和92.6%。上述实验数据表明,该 YOLOv7算法模型的改进策略是有效的。3.4对比实验为了进一步探明该算法与其他算法在船舶检测任务上的性能差异,本文在数据集上进行了对比实验。表3统计了对比算法模型的性能测试结果,可见,该算法模型mAP达到最高值为92.6%,比原始YOLOv7算法模型提高了 23.4%;检测速度为40.2 F s-1,能满足船舶图像检测技术实际应用中的耗时需求。此外,模型参数量大小仅为47.1 MB,这有利于在边缘设备进行部署。本文算法部署前的检测效果如图9所示,其中:G1为私人小船;G2

38、为极小的巡逻艇。现实中在内河航运上出行的船舶可能分布密集,这种情况利用现有大部分船舶图像检测技术均难以实现检测。部署后的检测结果如图10所示,其中:图10(a)中私人小船G1被正确检出为0,置信度达到0.7;巡逻艇G2被正确检出为2,置信度达到0.6;图10(b)中C1部分中的船舶也全部被成功检测。可见,针对小尺寸“货船”(标签为cargo ship,标签索引号为1)、“巡逻艇”(标签为patrol boat,标签索引号为2)和极小尺寸的“私人小船”(标签为personal boat,标签索引号为0)目标物,本文算法模型都具有良好的检测性能。(a)极小目标情况(b)密集性场景图9算法部署前的检

39、测结果举例Fig.9Research detection results before algorithm deployment G1 C1 G2(a)极小目标情况 (b)密集性场景图10算法部署后的检测结果举例Fig.10Research detection results after algorithm deployment表3不同对比算法模型的性能测试结果Table 3Performance test results of algorithm models compared目标检测方法SSD15Faster R-CNN16YOLOv317YOLOv418YOLOv5-L19YOLOX-M

40、20YOLOv7111优化算法模型mAP50/%53.364.962.364.865.068.569.292.6检测速度/(Fs-1)62.316.142.54349.750.151.640.2模型大小/MB92.3512.3221.4229.145.428.736.547.165厦门理工学院学报2024 年4结论本文提出一种内河航道中的船舶检测方法,通过负片图像增强算法来扩充少样本船舶标签数量,在YOLOv7模型算法基础上增加1个1010大小的细粒度检测头,并对检测头的锚框参数进行CIoU度量自适应优化:同时,在模型Backbone上引入SA注意力机制结构,完成船舶的目标检测。采用Cuda对

41、测试PC端运行加速,利用DNN推理方式部署模型后进行实验。测试结果显示:该方法全类别平均精度均值为92.6%,与原模型相比提高了23.4%;查准率和查全率分别达到92.8%和88.9%,与原模型相比分别提高了10.9%和23.6%;单张图片耗时仅需要32.26 ms。研究表明,该方法在准确性和实时性方面达到了平衡,能够高效应对其他物体遮挡和河中尾流等噪声干扰,对分布密集的小尺度船舶的检测亦依然有效。未来的工作将考虑采用其他有效策略进一步重新设计模型,以避免模型对背景中小目标产生的误检,并进一步提高船舶检测的准确率。另外,还将探索全尺度通用的蒸馏技术,以降低高分辨率图像的检测时长,提高效率。参考

42、文献1 吴喆.基于深度学习的动态背景下航道船舶检测识别与跟踪研究D.宜昌:三峡大学,2019.2 萧筝,王继业,夏叶亮.基于MSAM-YOLOv5的内河航道船舶识别方法J.华中科技大学学报(自然科学版),2023,51(5):67-73.3 PANG Y W,YUAN Y,LI X L,et al.Efficient HOG human detectionJ.Signal Processing,2011,91(4):773-781.4 NG P C,HENIKOFF S.SIFT:predicting amino acid changes that affect protein functio

43、nJ.Nucleic Acids Research,2003,31(13):3 812-3 814.5 薛远亮,金国栋,侯笑晗,等.融合注意力机制与改进SSD算法的SAR舰船目标检测方法J.计算机应用研究,2022,39(1):265-269.6 陈从平,吴喆,吴杞,等.动态背景下航道船舶目标检测方法J.三峡大学学报(自然科学版),2019,41(1):97-102.7 周薇娜,刘露.复杂场景下多尺度船舶实时检测方法J.电信科学,2022,38(10):67-78.8 李佳东,张丹普,范亚琼,等.基于改进YOLOv5的轻量级船舶目标检测算法J.计算机应用,2023,43(3):923-929.

44、9 周慧,李迎秋,陈澎,等.基于改进FPN的复杂场景下SAR图像船舶目标检测J.大连海事大学学报,2022,48(4):76-83.10 邓睿哲,陈启浩,陈奇,等.遥感影像船舶检测的特征金字塔网络建模方法J.测绘学报,2020,49(6):787-797.11 PATEL K,BHATT C,MAZZEO P L.Improved ship detection algorithm from satellite images using YOLOv7 and graph neural networkJ.Algorithms,2022,15(12):1-10.12 WANG Y C,LI J D,

45、CHEN Z M,et al.Ships small target detection based on the CBAM-YOLOX algorithmJ.Journal of Marine Science and Engineering,2022,10(12):1-8.13 XIE F,LIN B J,LIU Y C.Research on the coordinate attention mechanism fuse in a YOLOv5 deep learning detector for the SAR ship detection taskJ.Sensors,2022,22(9)

46、:3 370.14 ZHANG Q L,YANG Y B.Sa-net:shuffle attention for deep convolutional neural networksC/IEEE International Conference on Acoustics,Speech and Signal Processing(ICASSP).Toronto:IEEE,2021:2 235-2 239.15 马健,史文旭,鲍胜利.基于特征融合SSD的遥感图像舰船目标检测J.计算机应用,2019,39(增刊2):253-256.16 赵加坤,孙俊,韩睿,等.基于改进的Faster Rcnn遥感

47、图像目标检测J.计算机应用与软件,2022,39(5):192-196,290.17 HONG Z,YANG T,TONG X,et al.Multi-scale ship detection from SAR and optical imagery via a more accurate YOLOv3J.IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing,2021,14:6 083-6 101.18 JIANG J,FU X,QIN R,et al.High-speed lightw

48、eight ship detection algorithm based on YOLO-v4 for three-channels RGB SAR imageJ.Remote Sensing,2021,13(10):1 909.19 ZHANG J,ZHANG S,WANG S,et al.Research on target counting based on improved YOLOv5 and SORT algorithmsC/Proceedings of the 6th International Conference on Advances in Image Processing.Zhanjiang:Association for Computing Machinery,2022:109-113.20 LI S,FU X,DONG J.Improved ship detection algorithm based on YOLOX for SAR outline enhancement imageJ.Remote Sensing,2022,14(16):4 070.(责任编辑 雨松)66

展开阅读全文
相似文档                                   自信AI助手自信AI助手
猜你喜欢                                   自信AI导航自信AI导航
搜索标签

当前位置:首页 > 学术论文 > 论文指导/设计

移动网页_全站_页脚广告1

关于我们      便捷服务       自信AI       AI导航        获赠5币

©2010-2024 宁波自信网络信息技术有限公司  版权所有

客服电话:4008-655-100  投诉/维权电话:4009-655-100

gongan.png浙公网安备33021202000488号   

icp.png浙ICP备2021020529号-1  |  浙B2-20240490  

关注我们 :gzh.png    weibo.png    LOFTER.png 

客服