收藏 分销(赏)

基于深度学习的轻量级道路障碍检测算法.pdf

上传人:自信****多点 文档编号:2335855 上传时间:2024-05-28 格式:PDF 页数:12 大小:2.17MB
下载 相关 举报
基于深度学习的轻量级道路障碍检测算法.pdf_第1页
第1页 / 共12页
基于深度学习的轻量级道路障碍检测算法.pdf_第2页
第2页 / 共12页
亲,该文档总共12页,到这儿已超出免费预览范围,如果喜欢就下载吧!
资源描述

1、第46卷 第4期2023年8月Vol.46 No.4Aug.2023辽 宁 科 技 大 学 学 报Journal of University of Science and Technology Liaoning基于深度学习的轻量级道路障碍检测算法李正男,赵骥(辽宁科技大学 计算机与软件学院,辽宁 鞍山114051)摘要:道路障碍检测是智慧交通和无人驾驶的重要组成部分。针对道路障碍检算法的参数量过大、占用内存过多、难以在内存和算力有限的设备中使用等问题,本文提出一种轻量级的道路障碍检测算法。在YOLO v4的基础上,使用MobilenetV3作为模型的主干网络,减少模型参数,提高检测速度;改进模

2、型中的PAN结构,将主干网络中更浅层的特征图提取融合,改善小目标检测不佳的问题;在特征融合部分加入ECA注意力机制提升网络整体精度;提出一个新的DBR模块,使网络整体相比之前更加轻便。使用改进后的模型在自制数据集中进行检测,与Mobilenetv3-YOLO v4相比,精度提升5.24,参数量降低35.5%,F达43.8,满足实时应用的技术要求,表明模型可以嵌入到小型移动设备,达到良好的实时效果。关键词:路面障碍检测;轻量级模型;注意力机制;特征融合;深度可分离卷积中图分类号:TP399文献标识码:A文章编号:1674-1048(2023)04-0296-12DOI:10.13988/j.us

3、tl.2023.04.008路面障碍检测作为辅助驾驶功能,可以有效减少交通事故的发生,提高道路的维修效率。目前障碍检测中常用的技术有:毫米波雷达、激光测距仪和计算机视觉。Li等1利用三维激光雷达点云构建的网格图像提取障碍物信息,用于车道特征选择,避免行人和车辆的干扰。该方法利用自适应滑动窗口进行特征选择,利用分段最小二乘法进行道路线拟合,克服各种场景下光照变化、阴影遮挡、行人和车辆干扰等问题。Wang等2提出一种基于激光雷达的RGB图像辅助道路障碍物检测方法,通过增加点特征的数量弥补原始点云的稀疏性和不均匀性。Liu等3提出一种改进的轻量级实时语义分割网络,该网络基于一种高效的图像级联网络结构

4、,利用多尺度分支和级联特征融合单元提取丰富的多层次特征,提高了道路障碍物检测的准确性。但已有的路面障碍检测方法在目前的应用当中存在设备难以达到检测要求、成本过高、实时效果差和对小目标检测不明显等问题。针对以上问题,本文提出一种可实时应用的轻量级YOLO v4(You only look once version 4)模型,选用MobilenetV34作为主干网络,增加特征融合层,改善小目标的检测能力,引入ECA(Efficientchannel attention module)模块5提高模型的检测精度,并提出一个新的DBR轻量模块,使模型更加轻量化。1YOLO v4模型YOLO v46模型结

5、构如图1所示。在输入端使用马赛克数据增强,采用 cmBN(Cross mini-batch normalization)模块和 SAT(Self adversarialtraining)自对抗训练。主干网络使用 CSPDark-net53、Mish激活函数和正则化方法。在特征融合部分加入 SPP(Spatial pyramid pooling)7模块和PAN(Path aggregation network)8结构,在检测头中 使 用 CIOU_Loss(Complete intersection over收稿日期:2023-02-08。基金项目:国家自然科学基金资助项目(61775169);

6、辽宁省自然科学基金资助项目(2020-MS-281,20180551048);辽宁省教育厅科研项目(LJKZZ202-2043)。作者简介:李正男(1998),男,辽宁葫芦岛人。研究方向:机器学习、计算机视觉。通讯作者:赵骥(1974),男,辽宁鞍山人,教授。研究方向:机器学习、计算机视觉。第4期union loss)9作为损失函数,用 DIOU_nms(Dis-tance intersection over union non maximum suppres-sion)作为最后预测框的筛选。图片经过主干网络提取特征后,将不同尺度的特征图送入特征融合网络的 SPP 和 PAN 结构中,进行多尺

7、度的双向融合,最后在不同尺度的检测头中得到检测结果。1.1主干网络作为YOLO v4的主干网络,CSPDarknet53由五个CSP(Cross stage partial)10模块组合而成,每个CSP模块分别包含CBM(Conv batch normaliza-tion mish)模块和残差单元结构。其中残差单元解决了深层网络中存在的梯度消失和梯度爆炸问题,提升网络性能和效率。CSP与残差结构11的结合,解决梯度信息重复的问题,减少参数量的同时增强学习能力。1.2特征融合部分当主干网络完成特征提取后,模型使用特征融合12方式丰富卷积所得到的信息。在特征融合部分,YOLO v4使用SPP和PA

8、N结构进行融合操作,SPP结构可以增加网络对图片信息的感受野,PAN结构可以将不同层次的特征更加充分融合,并得到不同尺度的特征图。模型中的 PAN结构是在 FPN(Feature pyramid network)13结构的基础上添加一个自底向上的金字塔层,FPN将更高层次的语义特征向下传递,而 PAN 在此之上又加入自底向上的位置信息传递,使特征融合更加充分14。1.3检测头YOLO v4在三个特征尺寸不同的检测头中输出检测结果。在检测头中每个网格设置三个锚框,共有九个大小不同的锚框,将锚框和真实框的IOU(Intersection over union)以及分配的网格比较,将不同锚框划分为正

9、负样本。为了样本均衡,根据规则将部分锚框进行损失值计算,然后将真实框在网络中输出并计算损失值,最后得到分类、置信度、框位置和宽高的损失值,并加权求和。边界预测如图 2 所示。预测出偏移量值为(tx,ty,th,tw),根据坐标偏移计算式得到边界框的bx、by、bw、bh,最终输出结果。坐标偏移计算式bx=(tx)+cx(1)by=(ty)+cy(2)bw=pwetw(3)bh=pheth(4)图2边界预测框Fig.2Bounding prediction box图1YOLO v4结构图Fig.1Structure diagram of YOLO v4李正男,等:基于深度学习的轻量级道路障碍检测

10、算法 297辽 宁 科 技 大 学 学 报第46卷式中:cx和cy是特征图中单元网格的左上角坐标;pw和ph是预设的锚框映射到特征图中的宽和高;tx、ty、th、tw为预测出的偏移量坐标;为 Sig-moid函数。2模型改进为了减小网络计算量,使模型更加轻量化,本文使用MobilenetV3代替原本的主干网络进行特征提取操作。在特征融合网络中,丰富融合信息提升小目标检测15效果,加入ECA模块,有利于提高模型整体精度。提出新的DBR模块使模型整体更加轻便。改进后的模型结构如图3所示。2.1轻量级的特征提取网络由于YOLO v4的体积过大、参数过多,对检测设备要求较高,部署到实时设备中存在一些困

11、难,所以本文使用 MobilenetV3 作为模型的主干网络。MobilenetV3 是一个轻量级的网络,计算量小,推理时间短,十分适合路面障碍检测。2.1.1MobilenetV3整体结构MobilenetV3结构如图4所示。网络开头使用一个二维卷积进行操作。瓶颈模块使用线性瓶颈的逆残差结构处理数据。最后由一个批标准化的11的二维卷积,一个77的全局平均池化和两个11不使用批标准化的二维卷积组成,并在倒数第二个11的二维卷积中引入h-swish激活函数,提高模型效率。2.1.2瓶颈模块瓶颈模块结构如图5所示。在残差结构中使用深度可分离卷积和SE(Squeezeand excitation)通

12、道注意力机制。模块中使用11卷积,将低维空间映射到高维空间。在深度可分离卷积提取到足够多的信息后,进入SE注意力机制学习重要的数据特征。最后在不使用非线性激活函数的条件下再次使用11卷积降维。减小计算量的同时,不造成信息丢失。2.2ECA注意力机制在特征融合阶段使用ECA注意力机制提高模型整体检测精度,使模型有重点地学习数据特征。ECA注意力机制如图6所示。ECA注意力机制由全局平均池化、一维卷积和Sigmoid函数组图3改进后的模型结构Fig.3Improved model structure图4MobilenetV3结构图Fig.4Structure diagram of Mobilene

13、tV3 298第4期成。当不同尺度的特征图将要进入融合网络时,输入HWC大小的图像,数据经过全局平均池化层后得到尺度为11C的数据特征,之后使用卷积核为k的一维卷积,在不降维的基础上实现局部跨通道交互,减小模型参数,并且使用 Sig-moid函数使网络更加非线性化。2.2.1局部跨通道交互当特征由全局平均池化层输出进入一维卷积后,一维卷积没有对其进行降维操作,保持原来的通道大小。通道的关注点学习方式=(Wy)(5)式中:W是CC参数矩阵;y是聚合特征;是Sigmoid函数。模块中使用波段矩阵学习通道的关注点,捕获局部的跨通道交互。具体计算式i=j=1kwjiyjiyjiki(6)式中:yi的权

14、重只需考虑yi与其k个相邻的通道之间的相互作用,在一维卷积中权重共享,参数量从kC减小到k;kj表示yi相邻通道的集合。局部跨通道交互通过核大小为k的快速卷积实现,具体为=(C1Dk(y)(7)其中,C1Dk表示一个一维卷积。模块中应用这种方法,仅涉及到k个参数。2.2.2局部跨通道交互覆盖在局部通道交互中模块既实现了轻量化,又保证了局部跨通道交互的有效性。模块不仅可以自动确定交互的覆盖范围,还可以自适应调整各通道间的特征响应值。通道间的交互大小与通道的维度C成正比,在k与C之间存在映射C=(k)(8)其中,(k)为一个线性函数。线性映射是最简单的映射方法,但对于一些特征存在局限性。通道维数通

15、常设为2的指数倍,模型中采用2的指数函数表示非线性映射关系。在ECA中通过扩展线性函数(k),将其引入非线性函数C=(k)=2(k-b)(9)自动卷积核k的计算式为k=(C)=|log2(C)+bodd(10)图5瓶颈结构Fig.5Bottleneck structure图6ECA注意力机制Fig.6ECAattention mechanism李正男,等:基于深度学习的轻量级道路障碍检测算法 299辽 宁 科 技 大 学 学 报第46卷式中:|t|odd表示距离t最近的奇数;和b分别设置为2和1。2.3改进的特征融合层在网络特征提取阶段,层次较浅的网络提取出的浅层特征具有更加详细的细节信息,纹

16、理线条清晰,分辨率也更高。然而在更深层次的网络中,网络提取出的特征具有更大的感受野,可以得到更加丰富的语义信息。为了提高对小目标的检测效果,本文在特征融合阶段提取主干网络中更浅一层的特征图,并将其与原PAN结构中的特征相融合,使网络得到高分辨率特征和更加细微的数据信息,解决新特征层语义信息不够充分的问题。改进前后特征融合模型如图7所示。在原有模型基础上,加入一层新的特征,在主干网络中取出浅层特征,对尺度为104104的特征做卷积操作。将PAN中已经融合完成的特征做上采样,并将取出的104104特征图卷积后进行融合。进入CBL模块做5次卷积后下采样,开始做自上而下的融合操作。再与上采样前5252

17、的特征图做特征融合,并向下做卷积操作。原模型在1313、2626、5252这三层不同尺度大小特征之间进行相互融合(P13、P26、P52)。改进后的模型加入主干网络中更浅一层104104特征图进行融合操作(P13、P26、P52、P104)。充分利用四个不同尺度的特征层进行向上和向下的融合,使特征融合更加充分,增强信息间的交流。图7特征融合部分的改进Fig.7Improvement of feature fusion section2.4提出DBR轻量模块为了使模型整体更加轻便,本文采用深度可分离卷积、Relu6函数和批标准化技术,提出一个新的模块:DBR模块。2.4.1批标准化模型中网络输入

18、来自上一层的输出,并且网络通过反向传播的方式对参数进行调整,即在上一层网络参数发生变化前的输出过程中进行调整,这将导致网络参数的更新与输入的更新存在延迟。在网络层数多的模型中,每层网络的输入与权重值同时发生改变,这种训练比较困难。通常情况下要将学习率设置的很小,但这样将造成模型的学习速率十分缓慢。为了解决这些问题,本文在DBR模块中使用批标准化,使每一层输入中模型都对数据进行归一化处理。当模块经过卷积加入批标准化,可以有效防止网络发生梯度消失和过拟合问题。2.4.2深度可分离卷积深度可分离卷积过程如图8所示。深度可分离卷积包括逐通道卷积与逐点卷积,在逐通道卷积中每一个通道都对应一个特定的卷积核

19、,通道数不发生改变;逐点卷积使用11卷积对逐通道卷积的输出进行融合,并且随着卷积核的数量改变输出通道数量。假设将一个尺寸大小相同的特征图分别放入普通卷积和深度可分离卷积中,当特征图的尺寸为PkPkN,卷积核的大小为Pf PfN,卷积核的数量是M个,在特征图中的每个点都会做一次卷积的情况下,300第4期普通卷积的计算量为PkPkPfPfNM(11)深度可分离卷积的计算量为PkPkPfPfN+NMPkPk(12)显然,深度可分离卷积的计算量小很多。2.4.3模块设计本文提出由深度可分离卷积、批标准化和Relu6组合而成的轻量级模块。深度可分离卷积的特殊结构可大大降低模型的计算量。卷积后,使用批标准

20、化防止网络过拟合及梯度消失,并且加快收敛速度。在模块的末尾处使用Relu6作为模块的激活函数Relu=max(0,x)(13)Relu6=min(6,max(0,x)(14)当Relu6函数的x6时,函数导数为0。Re-lu6函数解决了精度损失问题。在本文的模型中,DBR模块被用于主干网络与特征融合网络的连接处和特征融合网络当中,与原来的CBL模块相比,模型整体变得更加轻便,并且整体精度也有所提升。DBR模块结构如图9所示。3网络训练和实验结果3.1路面障碍数据集本文进行路面障碍检测实验的数据集为自制数据集。确定被检测障碍的种类,使用百度、Google和个人拍摄等方式对数据进行收集,数据集中图

21、片统一为JPG格式。将路面障碍分为路障牌、减速带、路锥和坑洞四类,每个分类有500 张图图8深度可分离卷积Fig.8Depthwise separable convolution图9DBR模块Fig.9DBR module李正男,等:基于深度学习的轻量级道路障碍检测算法 301辽 宁 科 技 大 学 学 报第46卷片,共2 000 张,其中包括小目标、密集、遮挡和昏暗等情况,使用标签标注图片,标签格式为VOC格式。四类障碍的部分数据集如图10所示。3.2实验环境设备的CPU为Intel Xeon E5,内存128G,GPU为NVIDIAGTX TITAN XP,系统为Ubuntu 21.1。图

22、10部分数据集样本Fig.10Sample of partial data set实验过程中,使用冻结训练将主干网络冻结,使网络训练更加高效,保护权值。对实验参数进行调整,输入图片大小设置为416416,训练总轮次共100轮,前40轮设置为冻结训练,一次训练选取的样本大小为16,最大学习率为10-3;后60轮为解冻训练,一次训练选取的样本大小为4,最大学习率为10-4。将整个数据集分为训练集、验证集和测试集,其中,训练集占81%,验证集占9%,测试集占10%。模型训练的损失收敛曲线如图11所示。图中,A为加入改进的融合层,B为加入改进的融合层和ECA模块,C为加入改进的融合层、ECA模块并替换

23、DBR模块。使用预训练权重,模型在经过5轮后,损失值迅速降低。当模型解冻后,损失曲线在40到60轮明显下降,最后损失曲线趋于平稳。模型训练相对理想。图11不同模型的损失曲线图Fig.11Loss curves for different models3.3实验对比3.3.1评价指标使用准确率P、召回率R、平 302第4期均精度Ap、画面每秒传输帧数F和参数量这五项指标对模型进行评估。P、R、Ap计算公式P=TPTP+FP(15)R=TPTP+FN(16)Ap=01P(R)dR(17)式中:TP是正确的目标检测数量;FP是错误的目标检测数量;FN是错误的检测数量以及没有检测到的数量之和。P-R曲

24、线是准确率-召回率曲线。3.3.2轻量级模型的对比表1为不同轻量模型的对比实验在不同IOU阈值下的检测精度。本文使用MobilenetV3做特征提取。与MobilenetV1-YOLO v4相比,MobilenetV3参数量减少2.4%,精度上提高 1.31%。与 MobilenetV2-YOLO v4 相比,MobilenetV3参数量增加2.4%,但精度Ap,50明表1轻量级模型对比实验Tab.1Comparison experiment of lightweight models模型YOLO v4YOLO v4YOLO v4YOLO v4YOLO v4+YOLO v4+本文模型主干网络M

25、obilenetV1MobilenetV2MobilenetV3GhostnetMobilenetV1MobilenetV2MobilenetV3图片大小416416416416416416416416416416416416416416Ap,3086.3386.2486.2771.3686.5687.3887.62Ap,5073.4875.0274.7963.3975.576.9580.03Ap,7049.9250.7940.8032.6646.2450.8250.75参数量41 005 75739 124 54140 043 38939 740 43326 788 04924 900 17

26、125 819 019显提高。本文模型属于YOLO v4+系列。与YOLO v4系列模型相比,YOLO v4+系列模型参数量显著减少,十分轻便。三种模型的权重及F如表2所示。在检测速度方面,本文模型比YOLO v4的F快了8.4,检测权重减少154.2 M;比MobilenetV3-YOLO v4的F慢了2.3,但权重轻56.7 M。表2模型权重及FPSTab.2Weight and FPS of model模型CSPDarknet-YOLO v4MobilenetV3-YOLO v4本文模型权重/M256.9159.4102.7F35.446.143.83.3.3消融实验消融实验结果如表 3

27、 所示。为了提高模型对小目标的检测能力,本文增加更浅层的特征图,将详细的特征信息与丰富的语义信 息 相 融 合,Ap由 原 来 的 74.79%提 升 到77.73%。为了使模型的整体检测能力有所提高,模型中加入ECA模块增强模型学习特征的能力,Ap达到 80.01%。为了使模型整体更加轻量化,将网络中部分 CBL 模块替换成DBR模块,参数量降低 15 174 400,Ap也有 0.02%的提升。实验证明本文提出的方法是有效的。表3消融实验Tab.3Ablation experimentsMobilenetv3加入融合层ECADBRAp74.7977.7380.0180.03参数量40 04

28、3 38940 993 40540 993 41925 819 0193.3.4对比实验改进模型对不同四类路面障碍检测的P-R曲线如图12所示。改进后的模型对路障牌、减速带、路锥和坑洞的平均精准度分别达到:88.66%、75.98%、82.36%、73.11%,与Mobile-netV3-YOLO v4的检测精度相比,分别提升6.45、3.31、7.38和3.81。这表明改进后的模型整体精度都有提升。为了证明本文提出算法的有效性和先进性,李正男,等:基于深度学习的轻量级道路障碍检测算法 303辽 宁 科 技 大 学 学 报第46卷将本文模型与SSD 16、YOLO v3 17、Efficien

29、tDet 18、Centernet19、YOLO v4-Tiny20进行对比,结果如表4所示。本文模型的Ap,50达到80.03%,明显高于其他模型。图12四类路面障碍检测的P-R曲线Fig.12P-Rcurves for detecting four types of road obstacles表4对比实验Tab.4Comparative experiment模型SSDCenternetYOLO v3YOLO v4-TinyEfficientDet本文模型主干网络Vgg16Resnet50EfficientNetD0CSPdarknet-TinyEfficientnetD0Mobilene

30、tV3Ap,5069.6976.4278.1376.7277.7180.033.4效果图对比改进模型与原模型热力图检测结果如图 13所示。本文模型关注的路面障碍更加准确和完整,并且对多目标、小目标的检测感知准确,并精确定位。改进模型与原模型对路障牌、减速带、路锥、空洞的检测结果如图14所示。原模型存在漏检、图13热力图检测结果对比Fig.13Comparison of thermal map detection results 304第4期遮挡后难以检测等问题。本文模型在路面障碍数量拥挤与目标被遮挡的图片中,检测效果良好,对目标的定位十分准确,精度较高。4结论使用深度学习方法对道路中的障碍进行

31、检测,在提升检测速度的同时保证了对路面障碍检测的精度,使模型更方便地嵌入移动设备。(1)为了提高检测效率,使模型可以更好地实时应用,模型中使用MobilenetV3作为特征提取的主干网络,减小网络参数、提高检测速度,并且保证检测效果。(2)为了提高模型的精度,改善小目标检测的能力,在模型中增加特征融合层,将深层与浅层网络的优势互补,加入更加靠前的网络层,融合获得更加详细的特征信息,提升小目标和整体的检测精度。(3)在主干网络与特征融合网络的连接点加入ECA模块,有重点地学习不同通道特征,提高路面障碍检测的准确性。(4)在特征融合中加入DBR模块,模块由深度可分离卷积、批标准化和Relu6函数组

32、合而成,插入后使模型整体更加轻便。(5)与轻量级的MobilenetV3-YOLO v4模型相比,本文模型在精度上提升 5.24,参数量降低35.5%,F达43.8,满足实时应用的技术要求,提升了小目标检测效果,提升模型整体精度,使模型变得更加轻量化。图14道路障碍检测结果对比Fig.14Comparison of road obstacle detection results李正男,等:基于深度学习的轻量级道路障碍检测算法 305辽 宁 科 技 大 学 学 报第46卷参 考 文 献:1 LI J,SHI X X,WANG J Z,et al.Adaptive road detec-tion

33、method combining lane line and obstacle boundaryJ.IET Image Processing,2020,14(10):2216-2226.2 WANG Y Q,CHEN Y T,WU M L,et al.Fusing multi-modality information for 3D road obstacle detection C/International Symposium on Intelligent Signal Processingand Communication Systems(ISPACS),Taiwan,China,2021

34、:1-2.3 LIU B S,LV Y B,YANG G,et al.Implementation of alightweight semantic segmentation algorithm in road ob-stacle detection J.Sensors,2020,20(24):7089.4 HOWARD A G,PANG R M,ADAM H,et al.Search-ing for mobileNetV3C/IEEE/CVF International Con-ference on Computer Vision(ICCV),Seoul,Korea,2019:1314-13

35、24.5 WANG Q L,WU B G,ZHU P F,et al.ECA-net:effifi-cient channel attention for deep convolutional neural net-worksC/IEEE/CVF Conference on Computer Visionand Pattern Recognition(CVPR),Seattle,WA,USA,2020:11531-11539.6 LIN H B,ZHOU J L,CHEN M Z.Traffic sign detectionalgorithm based on improved YOLOv4

36、C/2022 IEEE10th Joint International Information Technology and Arti-ficial Intelligence Conference(ITAIC),Chongqing,China,2022:2156-2160.7 HE K M,ZHANG X Y,REN S Q,et al.Spatial pyra-mid pooling in deep convolutional networks for visualrecognition J.IEEE Transactions on Pattern Analysis&Machine Inte

37、lligence,2015,37(9):1904-1916.8 WANG W H,XIE E Z,SONG X G,et al.Efficient andaccurate arbitrary-shaped text detection with pixel aggre-gation network C/IEEE/CVF International Conferenceon Computer Vision(ICCV),Seoul,Korea,2019:8439-8448.9ZHENG Z H,WANG P,LIU W,et al.Distance-IoUloss:faster and bette

38、r learning for bounding box regres-sion C/Proceedings of the AAAI Conference on Artifi-cial Intelligence,USA:AAAI Press,2020:12993-13000.10WANG C Y,MARK LIAO H Y,WU Y H,et al.CSPNet:A new backbone that can enhance learning ca-pability of CNN C/IEEE/CVF Conference on Com-puter Vision and Pattern Reco

39、gnition(CVPR),Seattle,WA,USA,2020:1571-1580.11 HE K M,ZHANG X Y,REN S Q,et al.Deep residuallearning for Image recognitionC/IEEE Conferenceon Computer Vision and Pattern Recognition(CVPR),Las Vegas,NV,USA,2016:770-778.12CAO G M,XIE X M,YANG W Z,et al.Feature-fused SSD:fast detection for small objects

40、C/Pro-ceedings of the 9th International Conference on Graphicand Image Processing,Qingdao,China,2018:10615-10626.13LIN T Y,DOLLAR P,GIRSHICK R,et al.Featurepyramid networks for object detectionC/IEEE Con-ference on Computer Vision and Pattern Recognition(CVPR),Honolulu,HI,USA,2017:936-944.14 LIN M,Y

41、ANG D W,Pyramid frequency feature fusionobject detection networks J.Journal of Computer-Aid-ed Design and Computer Graphics,2021,33(2):207-214.15 ZHANG S Y,ZHANG Y.Small target pedestrian detec-tion based on multi-scale feature fusionJ.ComputerEngineering&Science,2019,41(9):1627-1634.16XU X H,ZHENG

42、H,GUO Z Y,et al.SDD-CNN:small data-driven convolution neural networks for sub-tle roller defect inspection J.Applied Sciences,2019,9(7):9071364.17 ZHANG X Z,JIANG Z,GUO W,et al.Research onthe influence of dehazing algorithm on YOLOv3 targetrecognitionC/2021 IEEE 4th International Confer-ence on Info

43、rmation Systems and Computer Aided Edu-cation(ICISCAE),Dalian,China,2021:471-475.18 TAN M X,PANG R M,LE Q V.EfficientDet:scalableand efficient object detectionC/2020 IEEE/CVFConference on Computer Vision and Pattern Recogni-tion(CVPR),Seattle,WA,USA,2020:10778-10787.19 ZHOU X Y,WANG D Q,KRAHENBUHL P

44、.Objectsas points J.https:/arxiv.org/pdf/1904.07850,2019-4-25.20WANG C Y,BOCHKOVSKIY A,LIAO H.Scaled-YOLOv4:Scaling cross stage partial networkC/2021 IEEE/CVF Conference on Computer Vision andPattern Recognition(CVPR),Nashville,TN,USA,2021:13029-13038.306第4期Lightweight road obstacle detection algori

45、thm based on deep learningLI Zhengnan,ZHAO Ji(School of Computer and Software Engineering,University of ScienceAnd Technology Liaoning,Anshan 114051,China)Abstract:Road obstacle detection is an important part of intelligent transportation and driverless driving.Forthe road obstacle detection algorit

46、hm,the number of parameters is too large,occupies too much memory,andis difficult to be used in equipment with limited memory and computing power.In this paper,a lightweightroad obstacle detection algorithm is proposed.Based on YOLO v4,MobilenetV3 is used as the backbone net-work of the model to red

47、uce model parameters and improve detection speed;The PAN structure in the im-proved model extracts and fuses shallower feature maps in the backbone network to improve the problem ofpoor detection of small targets;ECA attention mechanism is added to the feature fusion part to improve theoverall accur

48、acy of the network;A new DBR module is proposed to make the whole network more portablethan before.Using the improved model to detect in the self-made data set,compared with Mobilenetv3-YOLOv4,it not only improves the accuracy by 5.24%,but also reduces the number of parameters by 35.5%,andtheFalso r

49、eaches 43.8,which meets the technical requirements of real-time applications.The model can beembedded into a small mobile device to achieve good real-time effects.Keywords:road obstacle detection;lightweight model;attention mechanism;feature fusion;depthwise sep-arable convolution(Received February

50、8,2023)上接第261页Research progress on ceramic fibers for doping modificationof composite thermal barrier coatingsLI Yan,LIU Jiahang,LYU Zhe(School of Materials and Metallurgy,University of Science and Technology Anshan 114051,Liaoning,China)Abstract:With the continuous development of turbine engine tec

展开阅读全文
相似文档                                   自信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 

客服