基于改进YOLOv5s模型纸杯缺陷检测方法

蒋亚军, 曹昭辉, 丁椒平, 文煜超, 张闯, 胡志刚

包装工程(技术栏目) ›› 2023 ›› Issue (11) : 249-258.

PDF(4495 KB)
PDF(4495 KB)
包装工程(技术栏目) ›› 2023 ›› Issue (11) : 249-258. DOI: 10.19554/j.cnki.1001-3563.2023.11.029

基于改进YOLOv5s模型纸杯缺陷检测方法

  • 蒋亚军1, 曹昭辉1, 文煜超1, 张闯1, 胡志刚1, 丁椒平2
作者信息 +

Paper Cup Defect Detection Method Based on the Improved YOLOv5s Model

  • JIANG Ya-jun1, CAO Zhao-hui1, WEN Yu-chao1, ZHANG Chuang1, HU Zhi-gang1, DING Jiao-ping2
Author information +
文章历史 +

摘要

目的 针对小尺寸和特征不明显的纸杯缺陷在检测过程中易出现漏检、错检的问题,提出一种基于改进YOLOv5s模型纸杯缺陷检测方法。方法 在原始模型的Backbone部分引入CBAM注意力机制模块,提升模型的特征提取能力;增加一个YOLO检测头,将三尺度检测改为四尺度检测,提高模型对小目标和特征不明显目标的检测能力;在Neck部分借鉴加权双向特征金字塔网络BiFPN,对原始模型中的PANet进行部分改进,加强模型的特征融合能力。结果 结果显示,改进后的模型YOLOv5s–CXO精度为89.1%、召回率为90.4%、平均精度均值为89.5%,比原始模型的精度提高了1.5%、召回率提高了1.3%、平均精度均值提高了1.2%。结论 本文的改进方法有效提高了模型的检测能力,对小尺寸和特征不明显纸杯缺陷的检测效果有明显提升。

Abstract

The work aims to propose a paper cup defect detection method based on improved YOLOv5s model aiming at the problem that paper cup defects in small size and with insignificant features are easy to be missed and misdetected in the detection process. The CBAM attention mechanism module was introduced in the Backbone part of the original model to improve the feature extraction ability of the model. A YOLO detection head was added, and the three-scale detection was changed to four-scale detection to improve the detection ability of the model for small targets and targets with insignificant features. In the Neck part, the weighted bidirectional feature pyramid network BiFPN was used to partially improve the PANet in the original model to strengthen the feature fusion ability of the model. The results indicated that the precision of the improved model YOLOv5s-CXO was 89.1%, the recall was 90.4%, and the average precision mean was 89.5%. Compared with the original model, the precision was increased by 1.5%, the recall was increased by 1.3%, and the average precision mean was increased by 1.2%. The proposed improved method effectively improves the detection ability of the model, and significantly improves the detection effect of paper cup defects in small size and with insignificant features.

引用本文

导出引用
蒋亚军, 曹昭辉, 丁椒平, 文煜超, 张闯, 胡志刚. 基于改进YOLOv5s模型纸杯缺陷检测方法[J]. 包装工程(技术栏目). 2023(11): 249-258 https://doi.org/10.19554/j.cnki.1001-3563.2023.11.029
JIANG Ya-jun, CAO Zhao-hui, DING Jiao-ping, WEN Yu-chao, ZHANG Chuang, HU Zhi-gang. Paper Cup Defect Detection Method Based on the Improved YOLOv5s Model[J]. Packaging Engineering. 2023(11): 249-258 https://doi.org/10.19554/j.cnki.1001-3563.2023.11.029

基金

湖北省教育厅指导性项目(B2021119);武汉轻工大学校立科研项目(2021Y27)

PDF(4495 KB)

Accesses

Citation

Detail

段落导航
相关文章

/