基于改进YOLO v5的皮蛋裂纹在线检测方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金面上项目(32072302)和湖北省重点研发计划项目(2023BBB036)


Crack Detection Method for Preserved Eggs Based on Improved YOLO v5 for Online Inspection
Author:
Affiliation:

Fund Project:

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

    为了解决裂纹皮蛋分选中存在的效率低、人力成本高等问题,提出了一种基于改进YOLO v5的皮蛋裂纹在线检测方法。使用EfficientViT网络替换主干特征提取网络,并采用迁移学习对网络进行训练,分别得到YOLO v5n_EfficientViTb0和 YOLO v5s_EfficientViTb1两个模型。YOLO v5n_EfficientViTb0为轻量化模型,相较于改进前参数量减少14.8%,浮点数计算量减少26.8%;YOLO v5s_EfficientViTb1为高精度检测模型,平均精度均值为878%。采用GradCAM++对模型可视化分析,得出改进模型减少了对背景区域的关注度,证明了改进模型的有效性。设计了视频帧的目标框匹配算法,实现了视频中皮蛋的目标追踪,依据皮蛋的检测序列实现了对皮蛋的定位和裂纹与否的判别。轻量化模型的判别准确率为92.0%,高精度模型的判别准确率为943%。研究结果表明,改进得到的轻量化模型为运算能力较差的皮蛋裂纹在线检测装备提供了解决方案,改进得到的高精度模型为生产要求更高的皮蛋裂纹在线检测装备提供了技术支持。

    Abstract:

    With the aim to address the issues of low efficiency and high labor costs in crack detection and sorting of preserved eggs, a method for online crack detection based on an improved version of YOLO v5 was proposed. The backbone feature extraction network was replaced with the EfficientViT network, and the network was trained by using transfer learning, resulting in two models: YOLO v5n_EfficientViTb0 and YOLO v5s_EfficientViTb1. YOLO v5n_EfficientViTb0 served as a lightweight model, reducing the parameter size by 148% and the floating point operations by 268% compared with that of the original model. YOLO v5s_EfficientViTb1, on the other hand, was a high-precision detection model with an average precision mean of 878%. Through the utilization of GradCAM++ for model visualization and analysis, it was discovered that the improved model demonstrated a decreased focus on the background region. This finding served as evidence supporting the effectiveness of the enhancements implemented in the model. Moreover, a target box matching algorithm was designed for video frames to enable object tracking of preserved eggs in videos. Based on the detection sequence of preserved eggs, the algorithm achieved localization of the eggs and discrimination between cracked and intact ones. The lightweight model achieved a discrimination accuracy of 92.0%, while the high-precision model achieved an accuracy of 94.3%. These research findings indicated that the improved lightweight model provided a solution for preserved egg crack detection equipment with lower computational capabilities, while the improved high-precision model offered technical support for preserved egg crack detection equipment with higher production requirements.

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

汤文权,陈灼廷,王东桥,范维,王巧华.基于改进YOLO v5的皮蛋裂纹在线检测方法[J].农业机械学报,2024,55(2):384-392. TANG Wenquan, CHEN Zhuoting, WANG Dongqiao, FAN Wei, WANG Qiaohua. Crack Detection Method for Preserved Eggs Based on Improved YOLO v5 for Online Inspection[J]. Transactions of the Chinese Society for Agricultural Machinery,2024,55(2):384-392

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-06-19
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-02-10
  • 出版日期: