基于改进YOLOv8的烟卷污渍检测研究  

Research on Cigarette Stain Detection Based on Improved YOLOv8

作　　者：王翔 宗恩师 方贵阳 董晓泽 

机构地区：[1]红云红河烟草(集团)有限责任公司新疆卷烟厂,新疆 乌鲁木齐 [2]重庆科技大学计算机科学与工程学院,重庆

出　　处：《图像与信号处理》2024年第4期394-401,共8页Journal of Image and Signal Processing

基　　金：本文得到了重庆科技大学研究生科技创新计划项目“基于RESNET网络的烟卷污渍检测研究及实现”资助,编号ZNYKC2324。

摘　　要：在烟卷生产过程中,水渍、黄斑、褶皱和破损等缺陷是常见问题。传统的方法依赖于人工检测方法,但由于视觉疲劳等因素,导致生产效率不够高效。为了解决这一问题,本文提出了一种基于改进YOLOv8的烟卷污渍检测算法。本文以YOLOv8网络为骨干模型,加入了空间注意力机制,突出污渍区域的小目标检测能力,减小漏检和误检,从而减少小目标样本的误检率和漏检率。实验结果表明,与最新的YOLOv8算法相比,改进后的算法在烟卷污渍数据集上的平均精度均值(mAP)和召回率(Recall)分别提高了6.3%和1.8%;精度提高了6.2%。In the cigarette production process, defects such as water stains, yellow spots, wrinkles, and damage are common issues. Traditional methods rely on manual inspection, but due to factors like visual fatigue, production efficiency is not optimal. To address this problem, this paper proposes a cigarette stain detection algorithm based on an improved YOLOv8. We use the YOLOv8 network as the backbone model and incorporate a spatial attention mechanism to reduce the false detection and missed detection rates of small target samples. Experimental results show that, compared to the latest YOLOv8 algorithm, the improved algorithm increases the mean Average Precision (mAP) and Recall on the cigarette stain dataset by 6.3% and 1.8%, respectively, and improves precision by 6.2%.

关 键 词：YOLOv8 空间注意力机制 污渍检测 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]