CJS']JS-YOLOv5n: A high-performance detection model for cigarette appearance defects

被引:4
作者
Ma, Yihai [1 ,2 ]
Yuan, Guowu [1 ,2 ]
Yue, Kun [1 ,2 ]
Zhou, Hao [1 ,2 ]
机构
[1] Yunnan Univ, Sch Informat Sci & Engn, Kunming 650504, Peoples R China
[2] Yunnan Key Lab Intelligent Syst & Comp, Kunming 650504, Peoples R China
关键词
cigarette; defect detection; YOLOv5n; C2F; Jump Concat; SIoU;
D O I
10.3934/mbe.2023795
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
In tobacco production, cigarettes with appearance defects are inevitable and dramatically impact the quality of tobacco products. Currently, available methods do not balance the tension between detection accuracy and speed. To achieve accurate detection on a cigarette production line with the rate of 200 cigarettes per second, we propose a defect detection model for cigarette appearance based on YOLOv5n (You Only Look Once Version 5 Nano), called CJS-YOLOv5n (YOLOv5n with C2F (Cross Stage Partial (CSP) Bottleneck with 2 convolutions-fast), Jump Concat, and SCYLLAIoU (SIoU)). This model incorporates the C2F module proposed in the state-of-the-art object detection network YOLOv8 (You Only Look Once Version 8). This module optimizes the network by parallelizing additional gradient flow branches, enhancing the model's feature extraction capability and obtaining richer gradient information. Furthermore, this model uses Jump Concat to preserve minor defect feature information during the fusion process in the feature fusion pyramid's P4 layer. Additionally, this model integrates the SIoU localization loss function to improve localization accuracy and detection precision. Experimental results demonstrate that our proposed CJS-YOLOv5n model achieves superior overall performance. It maintains a detection speed of over 500 FPS (frames per second) while increasing the recall rate by 2.3% and mAP (mean average precision)@0.5 by 1.7%. The proposed model is suitable for application in high-speed cigarette production lines.
引用
收藏
页码:17886 / 17904
页数:19
相关论文
共 34 条
[1]  
Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, 10.48550/arXiv.2004.10934, DOI 10.48550/ARXIV.2004.10934]
[2]   SMD LED chips defect detection using a YOLOV3-dense model [J].
Chen, Ssu-Han ;
Tsai, Chia-Chun .
ADVANCED ENGINEERING INFORMATICS, 2021, 47
[3]   Research on Automatic Recognition of Casting Defects Based on Deep Learning [J].
Duan, Liming ;
Yang, Ke ;
Ruan, Lang .
IEEE ACCESS, 2021, 9 :12209-12216
[4]  
Gevorgyan Z, 2022, Arxiv, DOI [arXiv:2205.12740, 10.48550/arXiv:2205.12740]
[5]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[6]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[7]   The analysis of time-varying high-order moment of wind power time series [J].
Hao, Chen ;
Jin, Ting ;
Tan, FengLei ;
Gao, JinRui ;
Ma, ZhaoXing ;
Cao, Jing .
ENERGY REPORTS, 2023, 9 :3154-3159
[8]   Detection of PCB Surface Defects With Improved Faster-RCNN and Feature Pyramid Network [J].
Hu, Bing ;
Wang, Jianhui .
IEEE ACCESS, 2020, 8 :108335-108345
[9]   Uncertain barrier swaption pricing problem based on the fractional differential equation in Caputo sense [J].
Jin, Ting ;
Li, Fuzhen ;
Peng, Hongjun ;
Li, Bo ;
Jiang, Depeng .
SOFT COMPUTING, 2023, 27 (16) :11587-11602
[10]  
Jocher G. R., 2021, Ultralytics/YOLOv5: V5.0-YOLOv5-P61280 Models, AWS, Supervise.ly and YouTube Integrations