改进YOLOv5算法的钢筋端面检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金(62003191)


Steel-bar End Face Detection Based on Improved YOLOv5 Algorithm
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 增强出版
  • |
  • 文章评论
    摘要:

    钢筋是基建行业不可或缺的结构材料, 无论是钢筋生产过程、还是施工现场, 对钢筋进行准确计数是必不可少的环节. 成捆钢筋存在端面密集、直径尺度不统一、端面边界粘连、端面与背景融合、端面之间存在遮挡等问题. 针对上述问题, 提出了一种改进的YOLOv5模型框架, 以降低密集小目标漏检率、误检率. 针对钢筋端面数据集稀缺、没有公开的大型数据集并且钢筋端面特征较弱的问题, 自建了钢筋端面数据集, 使用半自动标注法对数据集进行标注, 并采用数据增强算法对钢筋端面数据集进行扩充. 修改了YOLOv5中的主干网络, 增加空间金字塔池(spatial pyramid pooling, SPP)和小目标检测层, 以获取更大的特征图; 使用特征金字塔模型(feature pyramid network, FPN)和路径聚合网络(path aggregation network, PAN)对多尺度特征图融合, 提高密集小目标检测精度. 在Data Fountain钢筋盘点竞赛数据集和自建钢筋数据集上设计了多组对照试验. 实验结果表明, 提出的改进算法YOLOv5-P2模型对钢筋端面的检测效果最佳, 钢筋端面平均精度均值(mean average precision, mAP)达到了99.9%, 相比于YOLOv3、YOLOv4、ScaledYOLOv4以及YOLOv5主流算法, 模型的mAP分别提升了9.6%、7.9%、7.0%、1.1%, 在工厂真实环境条件下进行测试时都有较稳定的表现, 在测试集上相对于原始模型检测精度提升了2.1%. 通过修改YOLOv5的主干网络中SPP模块位置和增加检测层都能够显著提升密集小目标检测精度, 更好的提取到钢筋端面的边缘特征, 取得99.9%的平均精度均值.

    Abstract:

    A steel bar is an indispensable structural material in the infrastructure industry, and accurate counting of steel bars is an essential link in both the steel-bar production process and the construction site. There are some problems in steel-bar bundles, such as dense end faces, non-uniform diameter scale, end-face boundary adhesion, fusion of end face and background, and end-face occlusion. To solve the above problems, this study proposes an improved YOLOv5 model framework to reduce the missed detection rate and the false detection rate of dense small targets. Considering the scarcity of the steel-bar end face dataset, the absence of a large public dataset in this field, and the weak feature of the steel-bar end face, we built a steel-bar end face dataset with the semi-automatic labeling method for dataset labeling and the data enhancement algorithm for dataset expansion. Moreover, the backbone network in YOLOv5 was modified, and the spatial pyramid pooling (SPP) and the small target detection layer were added to obtain larger feature maps. The feature pyramid network (FPN) and path aggregation network (PAN) were used to fuse multi-scale feature images to improve the accuracy of dense small target detection. Several groups of control tests were designed based on the Data Fountain steel-bar stocktaking competition dataset and the self-built steel bar dataset. The experimental results show that the improved algorithm YOLOv5-P2 model has the best performance on the steel-bar end face detection, and the mean average precision (mAP) of the steel-bar end face reaches 99.9%. Compared with the mainstream algorithms of YOLOv3, YOLOv4, ScaledYOLOv4, and YOLOv5, the proposed model has its mAP increased by 9.6%, 7.9%, 7.0%, and 1.1%, respectively. When tested in the real environment of factories, the model has stable performance, and its detection accuracy is improved by 2.1% compared with the original model on the test dataset. The position modification of the SPP module in the backbone network of YOLOv5 and the adding of detection layers can all significantly improve the detection accuracy of dense small targets with better edge feature extraction of the steel-bar end face and an mAP of 99.9%.

    参考文献
    相似文献
    引证文献
引用本文

王超,张运楚,孙绍涵,张汉元.改进YOLOv5算法的钢筋端面检测.计算机系统应用,2022,31(4):68-80

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2021-06-27
  • 最后修改日期:2021-07-02
  • 录用日期:
  • 在线发布日期: 2022-03-22
  • 出版日期:
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京海淀区中关村南四街4号 中科院软件园区 7号楼305房间,邮政编码:100190
电话:010-62661041 传真: Email:csa (a) iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号