首页|基于改进的YOLOv8n瓷砖缺陷检测与分类方法

基于改进的YOLOv8n瓷砖缺陷检测与分类方法

扫码查看
目前瓷砖表面的缺陷检测多数仍是由人工完成,生产效率低下,检测精度较低.为了解决以上问题,提出一种基于融入动态蛇形卷积和注意力机制的YOLOv8n瓷砖缺陷检测与分类方法.该方法首先在YOLOv8n模型主干网络和颈部网络中使用C2F-DSConv替换原来的C2F以增强对划痕缺陷特征的提取.其次,在模型中增加CBAM注意力模块以使模型能够更有效地从图像中学习到有用的特征,从而改善模型性能.最后,将损失函数CIoU修改为Inner-SIoU损失函数以增强模型对小目标的特征提取能力.实验结果表明,相较于原始模型,基于改进的YOLOv8n瓷砖缺陷检测模型的平均检测精度提高了2.6%且模型的计算量仅增加了0.5 G.此外,所提出的方法在对比实验和自建数据集实验中表现出明显的优势.
Defect Detection and Classification Method of Ceramic Tiles Based on Improved YOLOv8n
At present,most surface defect of ceramic tiles are still detected by workers with low production efficiency and low detection accuracy.In order to solve this problem,we proposed a ceramic tile defect detection and classification method based on YOLOv8n,with dynamic serpentine convolution and attention mechanism.Firstly,C2F-DSConv was used to replace the original C2F in the backbone network and neck network of the yolov8 model to enhance the extraction of scratch defect features.Secondly,the CBAM attention module was added to the model to enable it to learn useful features from the image more efficiently,so as to improve the model performance.Finally,the loss function CIoU was modified to the Inner-SIoU loss function to enhance the model's feature extraction ability for small targets.It is experimentally shown that,as compared with that of the original model,the average detection accuracy of the improved yolov8n-based tiles defect detection model is increased by 2.6%,while the number of the computation is only increased by 0.5 G.In addition,the proposed method showed obvious advantages in comparative experiments and self-built data set experiments.

ceramic tilesYOLOv8ndynamic snake convolutionattention mechanismsdefect detection

朱永红、吴松涛

展开 >

景德镇陶瓷大学机械电子工程学院,江西景德镇 333403

瓷砖 YOLOv8n 动态蛇形卷积 注意力机制 缺陷检测

2024

陶瓷学报
景德镇陶瓷学院

陶瓷学报

CSTPCD北大核心
影响因子:0.7
ISSN:1000-2278
年,卷(期):2024.45(6)