基于改进SSD的少样本目标检测.pdf
《基于改进SSD的少样本目标检测.pdf》由会员分享,可在线阅读,更多相关《基于改进SSD的少样本目标检测.pdf(6页珍藏版)》请在咨信网上搜索。
1、收稿日期:2022-12-07摇 摇 摇 摇 摇 摇 修回日期:2023-04-11基金项目:上海市地方能力建设项目(23010501500)作者简介:毕忠勤(1977-),男,博士,教授,研究方向为人工智能、计算机视觉;通讯作者:刘志斌(1989-),男,硕士,研究方向为计算机视觉。基于改进 SSD 的少样本目标检测毕忠勤1,单美静2,刘志斌1,徐富强1(1.上海电力大学 计算机科学与技术学院,上海 200090;2.华东政法大学 信息科学与技术系,上海 201620)摘摇 要:目标检测作为深度学习的热点问题之一,在自动驾驶、行人识别、智能医疗、机器人视觉等多个领域有着广泛的应用前景。但现有
2、的大部分目标检测模型都依赖于大规模的标注数据集来训练模型以保证目标检测的准确率,而在许多实际的应用场景中,大量数据的标注不仅耗费人力物力,而且需要大量专业人士的参与,在一定程度上限制了目标检测模型的实际应用。针对少样本目标检测的特殊要求,基于 SSD 网络提出了一种改进的少样本目标检测模型,提高了目标检测应用的适用性。首先,在 SSD(Single Shot multiBox Detector)网络的基础上,用 ResNet-50 代替 VGG 作为特征网络,从而提升模型的特征提取能力。其次,通过引入残差单元避免了网络退化问题。最后,为了充分融合各层之间的语义信息和位置信息,用 FPN(Fea
3、ture Pyramid Networks)替换了原模型中间的两个特征层。基于改进 SSD 网络的目标检测模型在少样本数据集的检测结果中,mAP 值达到了 79郾 8%,比原始模型提高了 2.6 百分点。关键词:目标检测;机器学习;少样本;FPN;SSD 网络模型中图分类号:TP393摇 摇 摇 摇 摇 摇 摇 文献标识码:A摇 摇 摇 摇 摇 摇 文章编号:1673-629X(2023)11-0035-06doi:10.3969/j.issn.1673-629X.2023.11.006Few-shot Object Detection Based on Improved SSDBI Zhon
4、g-qin1,SHAN Mei-jing2,LIU Zhi-bin1,XU Fu-qiang1(1.School of Computer Science and Technology,Shanghai University of Electric Power,Shanghai 200090,China;2.Department of Information Science and Technology,East China University ofPolitical Science and Law,Shanghai 201620,China)Abstract:As one of the ho
5、t topics of deep learning,object detection has a wide application prospect in many fields,such as automaticdriving,pedestrian recognition,intelligent medical treatment,and robot vision and so on.However,most of the existing object detectionmodels rely on large-scale annotation data sets for model tr
6、aining to ensure the accuracy of target detection.In many practical applicationscenarios,the annotation of a large number of data not only consumes human and material resources,but also requires the participation ofa large number of professionals,which limits the practical application of the object
7、detection model to a certain extent.Aiming at thespecial requirements of few-shot object detection,we propose an improved few-shot object detection model based on SSD(Single ShotmultiBox Detector)network,which improves the applicability of object detection applications.Firstly,the ResNet-50 is used
8、instead ofVGG as the feature network on the basis of SSD network to improve the feature extraction capability of the model.Secondly,the problemof network degradation is avoided by introducing residual element.Finally,in order to fully integrate the semantic information andlocation information betwee
9、n the layers,FPN is used to replace the two feature layers in the middle of the original model.In the detectionresults of the target detection model based on the improved SSD network in a few-shot data set,the mAP value of the improved modelreached 79.8%,which was 2.6 percentage points higher than t
10、hat of the original model.Key words:object detection;machine learning;few-shot;FPN;SSD0摇 引摇 言目标检测任务是计算机视觉领域最重要的任务之一,其研究主要用于人脸检测、安全系统、行人检测、自动驾驶、车辆检测以及医疗领域等1。在目前已知的研究领域中,要训练一个好的模型往往需要大量的数据,而人工智能令人印象深刻的特征是能够仅通过一个或几个样本就快速建立对新颖概念的认知的能力。这意味着当前的目标检测技术还有很大的发展空间。第 33 卷摇 第 11 期2023 年 11 月摇 摇 摇 摇 摇 摇 摇 摇 摇 摇计
11、算 机 技 术 与 发 展COMPUTER TECHNOLOGY AND DEVELOPMENT摇 摇 摇 摇 摇 摇 摇 摇 摇 摇Vol.33摇 No.11Nov.摇 2023为了提高目标检测应用的适用性,少样本目标检测任务成了当前研究的重点2。涉及到少样本目标检测的研究起步较早,早期的检测方法主要有基于模式匹配的方法,即使用小部分的样本作为模板,使用该样本在测试样本中寻找目标,后来也出现过形状匹配算法,即通过制定的标准来检测目标之间的相似性,然而传统的目标检测算法往往运算量大并且速度缓慢3。传统方法的特点是候选区域几何特征的选择,直线特征识别是早期图像目标的基本特征之一,其检测算法主要包
12、括:标准霍夫变换(SHT)4、Burns5方法和 LSD6方法等。由于霍夫变换对随机噪声和特征部分的遮挡具有较高鲁棒性,该方法得到了广泛的运用。而多边形特征识别是通过候选区域的各边界端点的图像二维坐标来进行描述,通过对各相邻特征点连线即可恢复完整的多边形信息。该方法的问题是,由于空间环境问题使得边界特征不完整,从而造成边界特征描述上的困难。传统的目标检测算法取得了很大的进展,但是仍然存在一些问题,例如,基于滑动窗口的区域选择策略没有针对性,时间复杂度高,窗口冗余,手工设计的特征对于环境多样性的变化并没有很好的鲁棒性。随着深度学习的发展,基于深度学习的少样本目标检测逐渐成为了研究热点。根据监督机
13、制的不同,基于深度学习的少样本目标检测算法主要分为有监督少样本目标检测、半监督少样本目标检测以及弱监督少样本目标检测。有监督少样本目标检测主要采用元学习方法、迁移学习方法或者数据增强方法。半监督少样本目标检测则主要基于自我训练或自我监督两种解决方案。弱监督少样本目标检测则主要包括基于主动对抗的方法、多实例学习方法以及基于度量学习的方法。该文对 SSD7网络运用于少样本目标检测任中存在的问题进行分析,优化 SSD 网络,优化后的模型不仅可以提高主干网络的特征提取能力,还可以有效提升对少样本数据集检测的准确性。首先将 SSD 网络中的主干网络替换成层数更深的 ResNet-50,高层次的网络模型有
14、效地提高了特征提取能力,并且利用残差单元避免了网络退化问题。其次,用 FPN 模块替换了原模型中的两个特征层,充分地融合了语义信息和位置信息。1摇 SSD 网络SSD 网络是传统的 one stage 目标检测算法,SSD的特点是生成的候选框都是相同尺寸,也是评分机制对候选框内的目标物体进行预测,最后使用非极大值抑制产生最终的检测结果。SSD 网络结构如图 1 所示。首先数据从输入层输入到模型中,接着由特征提取网络提取数据特征。SSD 的特征提取网络是由原始的 VGG16 网络和另外的五个卷积层构成,增强了对复杂特征的提取能力。从图中可以直观看出这些层的尺寸不一样,是为了实现在不同的尺度上对特
15、征图进行预测。Conv4_3Conv6Conv7Conv8_2Conv9_2Conv10_2Conv11_2VGG-16NMS图 1摇 SSD 网络模型摇 摇 SSD 算法中,提出了一个新的概念,即先验框,在特征提取网络生成特征图之后,以每个特征图为中心都会产生一系列不同尺度的先验框,这些先验框与原图一一对应,通过损失函数检测先验框的位置与真实目标框的位置的重叠度,从而分析出这个先验框内目标物体的类别,而且通过检测,也会对先验框的形状进行调整,以使其大小和尺度与真实的目标框更贴合。先验框包含两个参数:尺度 S 和宽高比 a。假设特征提取网络总共产生 m 个特征图,那么每个特征图的先验框尺度的计
16、算方式如公式(1)所示:Sk=Smin+Smax-Sminm-1(k-1),k 沂 1,m(1)SSD 算法对不同特征层生成特征图的尺度制定了如下规则:先验框的尺寸随着网络层数不断增加而增加,而特征图尺寸逐渐减小,不同特征层生成特征图的详细尺寸及先验框的大小如表 1 所示。表 1摇 每个特征图上的初始框尺度IndexFeaturemapFeaturemap sizeDefaultbox scaleTruesize1Conv6380.2602Conv7190.341023Conv8_2100.481444Conv9_250.621865Conv10_230.762286Conv11_210.92
17、7063摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 计算机技术与发展摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 第 33 卷摇 摇 对先验框的长宽比采用公式(2)的设置方式:ar沂 1,2,3,12,13(2)每个特征图上的初始框宽、高可以用公式(3)得出:wak=skar,摇 hak=sk/ar(3)对于宽高比为 1 的初始框,又多增加了一个正方形的初始框,定义如公式(4)所示。sk=sksk+1(4)SSD 所用到的特征图及其大小如表 2 所示。表 2摇 SSD 网络所用到的特征图及其大小feature mapSizeCo
18、nv4_338*38Conv719*19Conv8_210*10Conv9_25*5Conv10_23*3Conv11_21*1摇 摇 为了得到更好的检测结果,SSD 引入 ground truth的概念,即那些预测正确的样本。算法的关键是要搞清楚用哪个候选框进行预测,SSD 的操作是用先验框和训练样本中的 ground truth 进行比对,如果匹配成功,那与之对应的候选框就负责预测它。所以先验框和 ground truth 的匹配规则是算法是否有效的关键因素。SSD 的先验框与 ground truth 的匹配规则可以分成以下两个步骤:(1)对于图片中的众多候选目标,需要确定每个候选框对应
19、的 ground truth,计算不同先验框和 groundtruth 的 IOU,找出其中的最大值。这样就确定了一组匹配关系。因为 IOU 总有最大值,所以这种方式可以保证每个 ground truth 都不会被遗漏。如果某个先验框和 ground truth 成功匹配,称之为正样本。反之,该先验框对应的候选框定位失败,只能是负样本。(2)在第一轮筛选中剩余的先验框,不能全部舍弃,通过设定一个阈值,当某个 ground truth 与先验框的 IOU 大于这个阈值,那么也能达成匹配。但是这可能导致某个 ground truth 与多个先验框匹配成功,这是可以的,但是一个先验框只能匹配一个 g
20、round truth。2摇 改进 SSD 模型SSD 算法的训练已经非常成熟,因为它不需要生成候选框并且简化了特征提取的过程。通过实验表明,SSD 在 VOC2007 数据集上的检测精度达到了74郾 3%,比目前最先进的 Faster R-CNN 算法效果更好。与同类型的 one stage 算法相比,检测速度也更快。然而,SSD 仍然存在可以优化的地方:首先,该模型提出的先验框概念虽然有助于训练,但是先验框的初始尺度和长宽比需要手工设置,不能通过模型训练直接获得。而且 SSD 网络中每一层特征图使用的先验框的尺寸都不同,增加了训练的难度。其次,与Faster R-CNN 算法相比,SSD
21、对小样本的检测效果还存在明显的差距。因为 SSD 的底层特征图结构简单,无法对小目标的特征进行充分的提取。针对这些问题,该文对 SSD 算法进行优化,改进后的网络模型如图 2 所示,而对 SSD 算法提出的改进主要有:(1)将 SSD 的特征网络 VGG 替换成网络层数更深的 ResNet-50。增强对复杂特征的提取,并且利用残差单元,避免了网络层数加深带来的退化问题。(2)将 SSD 结构中 Conv8_2 和 Conv9_2 两个特征层替换成 FPN,更充分地融合不同层的语义信息,提升检测效果。Conv4_3Conv6Conv7FPNFPNConv10_2Conv11_2ResNet-50
22、NMS图 2摇 改进的 SSD 模型结构示意图摇 摇 下面对图 2 中的 ResNet-50、FPN 模块以及损失函数进行介绍。2.1摇 ResNet 模型ResNet8(残差网络)是由何凯明等提出的,解决了深层次网络模型退化的问题。ResNet 模型是在VGG19 模型的基础上进行了修改,该算法提出了残差单元这个概念。残差单元网络结构如图 3 所示。假设残差网络的输入为 X,将学习到的特征记为 H(X),使残差 F(X)=H(X)-X,那么就会得到原始的特征为F(X)+X。当残差为 0 时,即使使用多个网络堆积到73摇 第 11 期摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 摇 毕忠勤等:基
- 配套讲稿:
如PPT文件的首页显示word图标,表示该PPT已包含配套word讲稿。双击word图标可打开word文档。
- 特殊限制:
部分文档作品中含有的国旗、国徽等图片,仅作为作品整体效果示例展示,禁止商用。设计者仅对作品中独创性部分享有著作权。
- 关 键 词:
- 基于 改进 SSD 样本 目标 检测
1、咨信平台为文档C2C交易模式,即用户上传的文档直接被用户下载,收益归上传人(含作者)所有;本站仅是提供信息存储空间和展示预览,仅对用户上传内容的表现方式做保护处理,对上载内容不做任何修改或编辑。所展示的作品文档包括内容和图片全部来源于网络用户和作者上传投稿,我们不确定上传用户享有完全著作权,根据《信息网络传播权保护条例》,如果侵犯了您的版权、权益或隐私,请联系我们,核实后会尽快下架及时删除,并可随时和客服了解处理情况,尊重保护知识产权我们共同努力。
2、文档的总页数、文档格式和文档大小以系统显示为准(内容中显示的页数不一定正确),网站客服只以系统显示的页数、文件格式、文档大小作为仲裁依据,个别因单元格分列造成显示页码不一将协商解决,平台无法对文档的真实性、完整性、权威性、准确性、专业性及其观点立场做任何保证或承诺,下载前须认真查看,确认无误后再购买,务必慎重购买;若有违法违纪将进行移交司法处理,若涉侵权平台将进行基本处罚并下架。
3、本站所有内容均由用户上传,付费前请自行鉴别,如您付费,意味着您已接受本站规则且自行承担风险,本站不进行额外附加服务,虚拟产品一经售出概不退款(未进行购买下载可退充值款),文档一经付费(服务费)、不意味着购买了该文档的版权,仅供个人/单位学习、研究之用,不得用于商业用途,未经授权,严禁复制、发行、汇编、翻译或者网络传播等,侵权必究。
4、如你看到网页展示的文档有www.zixin.com.cn水印,是因预览和防盗链等技术需要对页面进行转换压缩成图而已,我们并不对上传的文档进行任何编辑或修改,文档下载后都不会有水印标识(原文档上传前个别存留的除外),下载后原文更清晰;试题试卷类文档,如果标题没有明确说明有答案则都视为没有答案,请知晓;PPT和DOC文档可被视为“模板”,允许上传人保留章节、目录结构的情况下删减部份的内容;PDF文档不管是原文档转换或图片扫描而得,本站不作要求视为允许,下载前自行私信或留言给上传者【自信****多点】。
5、本文档所展示的图片、画像、字体、音乐的版权可能需版权方额外授权,请谨慎使用;网站提供的党政主题相关内容(国旗、国徽、党徽--等)目的在于配合国家政策宣传,仅限个人学习分享使用,禁止用于任何广告和商用目的。
6、文档遇到问题,请及时私信或留言给本站上传会员【自信****多点】,需本站解决可联系【 微信客服】、【 QQ客服】,若有其他问题请点击或扫码反馈【 服务填表】;文档侵犯商业秘密、侵犯著作权、侵犯人身权等,请点击“【 版权申诉】”(推荐),意见反馈和侵权处理邮箱:1219186828@qq.com;也可以拔打客服电话:4008-655-100;投诉/维权电话:4009-655-100。