节点文献

一种基于YOLOv5s的改进装甲目标检测算法

An Improved Armored Target Detection Algorithm Based on YOLOv5s

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 易图明王先全袁威孔庆勇

【Author】 YI Tuming;WANG Xianquan;YUAN Wei;KONG Qingyong;Southwest Computer Co., Ltd.;Chongqing University of Technology;

【通讯作者】 王先全;

【机构】 西南计算机有限责任公司重庆理工大学

【摘要】 针对装甲目标图像背景复杂、目标尺度小等问题,提出一种基于YOLOv5s的装甲目标检测算法。首先在FPN结构中增加一个浅层分支,增强对小目标特征的提取能力;其次通过Focal Loss损失函数来平衡正负样本;再次将CIoU_loss用作边框回归损失函数,用以提升识别精度;最后将ECA注意力模块引入算法中,加强重要特征的表达。实验结果表明,改进算法在自制数据集上AP达到92.9%,相较于原始算法提高了4.2%,能够很好地满足装甲目标检测任务的精度与速度需求。

【Abstract】 Aiming at the problems of complex background and small target scale of armored target image, an armored target detection algorithm based on YOLOv5s is proposed. First, a shallow branch is added to the FPN structure to enhance the ability of extracting small target features; Secondly, the Focal Loss loss function is used to balance the positive and negative samples; CIoU_ Loss is used as the loss function of frame regression to improve the recognition accuracy; Finally, ECA attention module is introduced into the algorithm to enhance the expression of important features. The experimental results show that AP of the improved algorithm on the self-made data set achieves 92.9%, which is 4.2% higher than that of the original algorithm, and can well meet the accuracy and speed requirements of the armored target detection task.

  • 【文献出处】 现代信息科技 ,Modern Information Technology , 编辑部邮箱 ,2023年05期
  • 【分类号】TP391.41;E91
  • 【下载频次】84
节点文献中: 

本文链接的文献网络图示:

本文的引文网络