用C-CenterNet检测香烟外观缺陷,准确率达95.01%。
An Appearance Defect Detection Method for Cigarettes Based on C-CenterNet
- 基于关键点定位与属性回归,结合注意力机制增强特征提取。
- 在自动生成产线上实现95.01%的mAP,比原CenterNet提升6.14%。
- 适合工业质检场景,尤其对形状多变的缺陷有强适应性。
针对自动香烟生产线上传统方法适应性差、难以准确识别缺陷类型的问题,提出一种基于C-CenterNet的香烟外观缺陷检测方法。该检测器采用关键点估计定位中心点,并回归所有缺陷属性。首先使用ResNet50作为主干网络,引入卷积块注意力机制(CBAM)增强有效特征提取,减少非目标信息干扰;同时利用特征金字塔网络提升各层特征表达能力。其次,用可变形卷积替代部分常规卷积,增强对不同形状缺陷的学习能力。最后,以ACON激活函数替代ReLU,自适应选择部分神经元的激活操作,提升网络检测精度。实验结果主要基于平均精度均值(mAP)评估:C-CenterNet在香烟外观缺陷检测任务中mAP达95.01%,相比原始CenterNet模型提升6.14%,满足自动生产线对精度与适应性的要求。
原文摘要 · Abstract (English)
Due to the poor adaptability of traditional methods in the cigarette detection task on the automatic cigarette production line, it is difficult to accurately identify whether a cigarette has defects and the types of defects; thus, a cigarette appearance defect detection method based on C-CenterNet is proposed. This detector uses keypoint estimation to locate center points and regresses all other defect properties. Firstly, Resnet50 is used as the backbone feature extraction network, and the convolutional block attention mechanism (CBAM) is introduced to enhance the network's ability to extract effective features and reduce the interference of non-target information. At the same time, the feature pyramid network is used to enhance the feature extraction of each layer. Then, deformable convolution is used to replace part of the common convolution to enhance the learning ability of different shape defects. Finally, the activation function ACON (ActivateOrNot) is used instead of the ReLU activation function, and the activation operation of some neurons is adaptively selected to improve the detection accuracy of the network. The experimental results are mainly acquired via the mean Average Precision (mAP). The experimental results show that the mAP of the C-CenterNet model applied in the cigarette appearance defect detection task is 95.01%. Compared with the original CenterNet model, the model's success rate is increased by 6.14%, so it can meet the requirements of precision and adaptability in cigarette detection tasks on the automatic cigarette production line.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。