过度拟合
卷积神经网络
计算机科学
人工智能
特征(语言学)
算法
棱锥(几何)
深度学习
特征提取
人工神经网络
模式识别(心理学)
卷积(计算机科学)
一般化
过程(计算)
数据挖掘
数学
语言学
操作系统
数学分析
哲学
几何学
作者
Dehua Zhang,Xinyuan Hao,Linlin Liang,Wei Liu,Chunbin Qin
出处
期刊:Journal of Computational Design and Engineering
[Oxford University Press]
日期:2022-07-29
卷期号:9 (5): 1616-1632
被引量:37
摘要
Abstract The surface defect detection (SDD) problem is one of the crucial techniques during production process, so it has become a key research area to control the quality of industrial products, which has been increasingly of greater interest to the researchers especially with the rapid development of artificial neural networks technology in recent years. Therefore, this paper proposes a novel deep convolutional neural network algorithm aiming at SDD. Firstly, a dense cross-stage partial Darknet backbone network is designed for feature extraction by optimizing cross-stage partial Darknet through the idea of dense connections, which can, not only enhance feature reuse but also greatly alleviate the overfitting issue. Secondly, a new cross-stage hierarchy module is presented combining the cross-stage feature fusion strategy and depthwise separable convolution technique for each node of the path aggregated feature pyramid network (PAN). Finally, an efficient channel attention (ECA) mechanism is introduced in PAN to construct a novel ECA PAN. The experimental results on three surface defect datasets show that the mean average precision of this network is 2.63, 5.48, and 1.16$\%$ which is higher than that of the baseline network, respectively. The proposed network outperforms not only the classical models but state-of-the-art models, which indicates the proposed algorithm can achieve higher accuracy and speed with fewer calculation parameters. And what is more, the proposed algorithm also has outstanding generalization ability.
科研通智能强力驱动
Strongly Powered by AbleSci AI