吴维崧,涂福泉,罗迎九,杨家瑜,韩天宇,汪曙峰,涂楚杰.融合注意力机制的弱监督纸板表面缺陷检测[J].包装工程,2024,45(3):201-207.
WU Weisong,TU Fuquan,LUO Yingjiu,YANG Jiayu,HAN Tianyu,WANG Shufeng,TU Chujie.Weakly Supervised Cardboard Surface Defect Detection with Attention Mechanism[J].Packaging Engineering,2024,45(3):201-207. |
| 融合注意力机制的弱监督纸板表面缺陷检测 |
| Weakly Supervised Cardboard Surface Defect Detection with Attention Mechanism |
| 投稿时间：2023-06-06 |
| DOI：10.19554/j.cnki.1001-3563.2024.03.023 |
| 中文关键词: 弱监督学习 对象定位 深度学习 纸板表面缺陷检测 自注意力 |
| 英文关键词: weakly supervised algorithm object localization deep learning cardboard surface defect detection self-attention |
| 基金项目:国家自然科学基金(51701145) |
|
| 摘要点击次数: |
| 全文下载次数: |
| 中文摘要: |
| 目的 针对目前表面缺陷检测方法因缺少实例级标签，使深度神经网络在工业检测上的应用受到限制的问题。本文面向实际的纸板表面缺陷检测任务，提出弱监督学习下融合卷积和注意力机制的神经网络算法。方法 该网络通过将通道注意力模块和梯度类激活映射模块相结合，进一步提高类激活图的精细度，实现纸板表面缺陷的精确定位;同时通过倒残缺结构和上采样层的组合操作，进一步细化浅层特征提升网络的特征提取能力，加快网络收敛速度。结果 通过在公开的纸板缺陷数据集上进行实验，本文提出的算法在使用图像级标签训练的情况下，分类正确率与定位正确率分别达到99.0%和92.2%，验证了该算法的有效性。结论 避免了实例级标签数量较少和过于主观的缺点，为基于机器人的缺陷纸板剔除奠定了基础。 |
| 英文摘要: |
| The application of deep neural networks in industrial inspection is limited due to the lack of instance-level labels. To address this issue, the work aims to propose a neural network algorithm that combines convolution and attention mechanisms under weakly supervised learning for practical surface defect detection on cardboard. By integrating channel attention modules and gradient-based activation mapping modules, this network enhanced the precision of class activation maps and realized the precise localization of cardboard surface defects. Additionally, a combination of inverted residual structures and upsampling layers was utilized to refine shallow features and improve the network's feature extraction capabilities, thereby accelerating the convergence speed. Experiments were carried out on the publicly available cardboard defect dataset, achieving classification accuracy and localization accuracy of 99.0% and 92.2% respectively under the training with image-level labels and demonstrating the effectiveness of the proposed algorithm. The disadvantages of a small number of instance-level labels and excessive subjectivity are avoided, which lays a foundation for the removal of defective cardboard based on robots. |
|
查看全文
查看/发表评论 下载PDF阅读器 |
| 关闭 |
|
|
|
渝公网安备 50010702501716号