X射线图像违禁品自动检测算法研究.pdf
《X射线图像违禁品自动检测算法研究.pdf》由会员分享,可在线阅读,更多相关《X射线图像违禁品自动检测算法研究.pdf(3页珍藏版)》请在咨信网上搜索。
1、随着社会的不断发展,城市轨道交通在人们生活中占据着越来越重要的作用。据统计,广州地铁客运量排在全国首位,2022 年达到 231 874 万人。面对如此巨大的客运量,安检的自动化、实时性、高准确率是很有必要的1。传统的目标检测算法面对背景庞杂、行李乱放、违禁品多种多样等情况,能力不足,难以满足客运的要求。2012 年,Alex Krizhevsky 等2设计的AlexNet 在 ImageNet 挑战赛上夺得冠军,大幅提升了图像分类的准确度,深度学习重新迎来春天,开始在各个领域开花结果,但在安检领域的应用相对较少。YOLO 系列3是目标检测领域知名度最高的算法,其凭借出色的实时检测性,在不同的
2、领域均有广泛应用。YOLO 系列算法将问题概括为一个回归问题,一次完成,是一种端对端的卷积神经网络。本文选择 YOLOv5m 算法来进行安检违禁品识别,并增加置换注意力(Shuffle Attention,SA)模块4,以提高检测准确率。1YOLOv5 算法YOLOv5 算 法 主 要 包 含 Input(输 入 端)、Backbone(主干网络)、Neck(多尺度特征融合网络)和 Prediction(检测头)5。Input 部分是行李经过X 射线安检后采集到的图像;Backbone 部分包含多种卷积、池化、归一化、激活函数等操作,主要是进行图像的特征提取;Neck 部分主要是进行多尺度特征
3、融合,增加主干特征的可接受性,丰富上下文信息;Prediction 部分采用分类、定位、置信度共3 个分支,用于获取目标的类别、位置和置信度。YOLOv5m 算法结构见图 1。Backbone 主要由CBS(Conv+BatchNorm+SiLU)、跨阶段局部(CrossStage Partial,CSP)、共享压缩分析森林(SharedPacked Parse Forest,SPPF)结构组成,CBS 由卷积、归一化、SiLU 激活函数构成。Backbone 采用的是带有残差(Res)组件的 CSP1 结构。Neck 包括 特 征 金 字 塔 网 络(Feature Pyramid Netw
4、orks,FPN)和路径聚合网络(Path Aggregation Network,PAN)结构设计,采用的是不带 Res 组件的 CSP2结构。目标检测任务的损失函数由分类损失、定位损失和置信度损失的加权得到。在训练过程中,输入图像采用 Mosaic 数据增强、自适应锚框计算、自适应图片缩放等方式,提高了模型的泛化能力,减少了计算量。基金项目 山西省高等学校科技创新项目(2022L707);长安大学中央高校基本科研业务费专项资金资助项目(300102213515)收稿日期:2023原04原06曰修回日期:2023原04原30作者简介:令狐蓉(1989),女,山西运城人,硕士,助教,主要从事城
5、市轨道交通通信与信号研究,E-mail:。X 射线图像违禁品自动检测算法研究令狐蓉摘要:为了提高城市轨道交通 X 射线安检违禁品自动识别的有效性、提升安检效率,设计了一种带有注意力机制的YOLOv5m 算法模型。该算法模型引入置换注意力(Shuffle Attention,SA)模块,采用通道随机混合操作,分块并行使用空间和通道两类注意力机制,使两类注意力机制高效结合。通过实验验证,改进后的算法的 mAP50 达到了 0.718,相较于基础的 YOLOv5m 算法,mAP50:95 和 mAP50 分别提升了 0.8 个百分点和 1.2 个百分点,结果表明改进后的算法能显著提高检测识别精度。关
6、键词:YOLOv5;SA 模块;X 射线;识别;违禁品中图分类号:TP301.6;O434.19;TP391.41文献标识码:ADOI:10.3969/j.issn.1674-9146.2023.09.105(山西工程职业学院交通工程系,山西太原030001)文章编号:1674-9146渊圆园23冤09原105原03科 技 创 新 与 生 产 力SCI-TECH INNOVATION&PRODUCTIVITY第 44卷第 9 期2023年 9 月Vol.44No.9Sep.2023*图 1YOLOv5m 算法结构图科技创新与生产力 2023年第 44卷第 9 期算法模型mAP50:95mAP5
7、0ATSS+LAreg0.5850.709ATSS+LAcls0.5930.718YOLOv5m0.6050.706YOLOv5m+SA0.6130.7182YOLOv5 算法的改进南京大学杨育彬教授等提出 SA 模块4,高效地结合这两种注意力机制,并引入了特征分组与通道置换,得到了一种超轻量型的注意力机制。SA 模块结构见图 2,它采用置换单元,高效组合上述两种类型的注意力机制。首先将输入沿着通道维度拆分为多组,然后对每一组特征词用置换单元刻画特征在空间和通道两个维度上的依赖性,最后所有特征进行集成并通过通道置换操作进行组件特征合并6。本文将 SA 模块嵌入到 Prediction 部分之前
8、,即在 CSP2 和 Conv 之间添加,改进后的 YOLOv5m 算法的 Prediction 部分结构见图 3。3实验结果与分析3.1数据集本文采用赵才荣教授公开的刀具和液体容器 X射线(Cutters and Liquid Containers X-ray,CLCX-ray)数据集7。CLCXray 数据集包含 9 565 张 X 射线安检图像,包括 5 种刀具和 7 种液体容器,共有 12个类别。5 种刀具包括刀片、匕首、刀、剪刀、瑞士军刀,7 种液体容器包括易拉罐、纸盒饮料、玻璃瓶、塑料瓶、真空杯、喷罐、锡罐。3.2评价指标本文采用 mAP50:95和 mAP50作为算法模型的评价指
9、标,平均精度均值(mean Average Precision,mAP)是准确率-召回率(Precision-Recall,P-R)曲线下的面积。1)mAP50:95。在不同阈值(从 0.5 到 0.95,步长 0.05)下的 mAP。2)mAP50。计 算 每 一 类 别 的 交 并 比(Intersection over Union,IoU)设为 0.5 时的平均精度(Average Precision,AP),对所有类别的 AP 求平均值。3.3实验结果YOLOv5m 算法模型的性能及其加入 SA 模块后的 YOLOv5m 算法模型的性能实验结果见表 1,同时与赵才荣教授团队提出的 AT
10、SS 模型的性能进行对比。从表 1 中可以看出,YOLOv5m 算法模型比ATSS+LAreg 算法模型和 ATSS+LAcls 算法模型在mAP50:95上分别提升了 2 个百分点和 1.2 个百分点;而 YOLOv5m 算法模型在添加 SA 模块后,mAP50:95和 mAP50分别提升了 0.8 个百分点和 1.2 个百分点。结果表明,基于 SA 模块改进的 YOLOv5m 算法模型在 CLCXray 数据集中的违禁品检测识别中有一定的优越性。图 4 为检测识别效果图。4结论针对城市轨道交通 X 射线安检违禁品的检测识别,本文在 YOLOv5m 算法基础上,增加了 SA 模块,分别对空间
11、和通道的特征进行操作,经过在CLCXray 数据集上进行实验验证,结果表明改进后的算法能显著提高检测识别精度。但是改进后的算法在易拉罐、玻璃瓶、喷罐等类别上的检测识别效果不理想,下一步研究可以围绕易拉罐、玻璃瓶、喷罐 3 个类别的图像进行改进。参考文献:1任健.基于改进 YOLOv5 的轻量化违禁物检测技术研究与实践D.成都:四川师范大学,2022.图 2SA 模块结构图图 3改进后的 YOLOv5m 算法的 Prediction 部分结构图表 1不同算法模型的性能对比图 4检测识别效果图PredictionSAConvSASAConvConvC8 106 Research on Automa
- 配套讲稿:
如PPT文件的首页显示word图标,表示该PPT已包含配套word讲稿。双击word图标可打开word文档。
- 特殊限制:
部分文档作品中含有的国旗、国徽等图片,仅作为作品整体效果示例展示,禁止商用。设计者仅对作品中独创性部分享有著作权。
- 关 键 词:
- 射线 图像 违禁品 自动检测 算法 研究
1、咨信平台为文档C2C交易模式,即用户上传的文档直接被用户下载,收益归上传人(含作者)所有;本站仅是提供信息存储空间和展示预览,仅对用户上传内容的表现方式做保护处理,对上载内容不做任何修改或编辑。所展示的作品文档包括内容和图片全部来源于网络用户和作者上传投稿,我们不确定上传用户享有完全著作权,根据《信息网络传播权保护条例》,如果侵犯了您的版权、权益或隐私,请联系我们,核实后会尽快下架及时删除,并可随时和客服了解处理情况,尊重保护知识产权我们共同努力。
2、文档的总页数、文档格式和文档大小以系统显示为准(内容中显示的页数不一定正确),网站客服只以系统显示的页数、文件格式、文档大小作为仲裁依据,平台无法对文档的真实性、完整性、权威性、准确性、专业性及其观点立场做任何保证或承诺,下载前须认真查看,确认无误后再购买,务必慎重购买;若有违法违纪将进行移交司法处理,若涉侵权平台将进行基本处罚并下架。
3、本站所有内容均由用户上传,付费前请自行鉴别,如您付费,意味着您已接受本站规则且自行承担风险,本站不进行额外附加服务,虚拟产品一经售出概不退款(未进行购买下载可退充值款),文档一经付费(服务费)、不意味着购买了该文档的版权,仅供个人/单位学习、研究之用,不得用于商业用途,未经授权,严禁复制、发行、汇编、翻译或者网络传播等,侵权必究。
4、如你看到网页展示的文档有www.zixin.com.cn水印,是因预览和防盗链等技术需要对页面进行转换压缩成图而已,我们并不对上传的文档进行任何编辑或修改,文档下载后都不会有水印标识(原文档上传前个别存留的除外),下载后原文更清晰;试题试卷类文档,如果标题没有明确说明有答案则都视为没有答案,请知晓;PPT和DOC文档可被视为“模板”,允许上传人保留章节、目录结构的情况下删减部份的内容;PDF文档不管是原文档转换或图片扫描而得,本站不作要求视为允许,下载前自行私信或留言给上传者【自信****多点】。
5、本文档所展示的图片、画像、字体、音乐的版权可能需版权方额外授权,请谨慎使用;网站提供的党政主题相关内容(国旗、国徽、党徽--等)目的在于配合国家政策宣传,仅限个人学习分享使用,禁止用于任何广告和商用目的。
6、文档遇到问题,请及时私信或留言给本站上传会员【自信****多点】,需本站解决可联系【 微信客服】、【 QQ客服】,若有其他问题请点击或扫码反馈【 服务填表】;文档侵犯商业秘密、侵犯著作权、侵犯人身权等,请点击“【 版权申诉】”(推荐),意见反馈和侵权处理邮箱:1219186828@qq.com;也可以拔打客服电话:4008-655-100;投诉/维权电话:4009-655-100。