轻量化模型的PeleeNet_yolov3地表裂缝识别
CSTR:
作者:
作者单位:

(四川大学 电气工程学院,成都 610065)

作者简介:

刘星(1997—),男,硕士研究生;张江(1973—),男,副教授,博士生导师

通讯作者:

莫思特,1400551690@qq.com

中图分类号:

TP391.4

基金项目:

国家重点研发计划(2018YFC1505502)


PeleeNet_yolov3 surface crack identification with lightweight model
Author:
Affiliation:

(College of Electrical Engineering, Sichuan University, Chengdu 610065, China)

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为提高地表裂缝检测在低算力运算平台上的稳定性和检测速率,提出了一种PeleeNet与YOLOv3相结合的目标检测算法。使用PeleeNet框架代替YOLOv3的Darknet-53主体框架,以融合不同的局部特征及提高运算效率;在框架中融合特征注意力模块以提高图像中裂缝区域的显著度,并通过对感受野模块RFB卷积核的复用,增大网络的有效视野,提高小目标检测精度;在特征金字塔网络中,通过使用深度可分离卷积代替标准卷积,减少参数计算量;引入CIoU损失函数提高模型的分类与回归精度。在测试平台上应用裂缝数据进行算法验证,结果表明:AP50达到了97.68%,AP75达到了77.87%,较原始的YOLOv3分别提高8.4%和12.4%,检测速度达到了30帧/s,且模型参数大小仅为原始YOLOv3的30%;可以看出,本研究提出的PeleeNet_yolov3轻量化模型对于裂缝目标的检测效果较为明显,并且具有较小的运算量和参数量,适合应用于移动端系统,对于小体积低功耗低算力运算平台具有较大应用价值。

    Abstract:

    To enhance the stability and speed of surface crack detection, a detection algorithm combining PeleeNet and YOLOv3 is proposed. PeleeNet framework is used to replace Darknet-53 framework of YOLOv3 so as to effectively integrate different local features and improve the detection rate. The feature attention module is integrated into the PeleeNet’s framework to highlight the saliency of the crack detection in the image, and through the receptive field module RFB broaden the effective field of view of the network, and increase the detection accuracy of small targets. Instead of the standard convolution, the depth separable convolution is employed to reduce the amount of parameter calculation in the feature pyramid network. Then, the CIoU loss function is introduced to strengthen the classification and regression accuracy of the model. The experimental results on the fracture data set show that AP50 and AP75 reach 97.68% and 77.87% respectively, 8.4% and 12.4% higher than the original YOLOv3. Meanwhile, the detection speed reaches 30 frames per second with the size of model parameters being only 30% of YOLOv3. As can be seen, the PeleeNet_yolov3 lightweight model proposed has produced an obvious effect on the detection of crack targets, with a small amount of calculations and parameters involved. Suitable for mobile terminal system, the study presents a great value of application especially for small volume, low power consumption and low computing power computing platforms.

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

刘星,莫思特,张江,王炜康,杨世基,李鑫.轻量化模型的PeleeNet_yolov3地表裂缝识别[J].哈尔滨工业大学学报,2023,55(4):81. DOI:10.11918/202112015

复制
相关视频

分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2021-12-03
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-04-10
  • 出版日期:
文章二维码