.png){kind=icon}
返回
Surface Defect Detection Algorithm Based on Feature-Enhanced YOLO
.png){kind=icon}
PRE
AI
DOI：10.1007/s12559-022-10061-z.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Surface defect detection is a complicated task to achieve both specific class and precise location of each defect. Specifically for industrial scenario, realizing efficient and accuracy-satisfactory surface defect automatic detection is still a big challenge. Therefore, a surface defect detection algorithm based on feature-enhanced YOLO (FE-YOLO) for practical industrial applications is proposed in this paper. For the purpose of efficient detection, we lighten YOLO model by combining deep separable convolution and dense join. And an improved feature pyramid network is proposed to enhance the spatial location correlation for multi-scale detection layer for the sake of high accuracy. Then, a new loss function of prediction box regression is established to boost the detection accuracy under the high Intersection over Union (IoU) threshold and accelerate model convergence. To select anchor boxes of different scale feature detection layers, we propose a statistical-based k-means++ algorithm, which can improve the quality of initial anchors and accelerate the convergence of the proposed model. Two industrial surface defect datasets, NEU-DET dataset and DeepPCB dataset, are used to verify the effectiveness of the proposed FE-YOLO algorithm. Experimental results demonstrate that FE-YOLO algorithm is lightened nearly 80% compared with YOLOV4. The detection speed is better than the other state-of-the-art surface defect detection algorithms. The defects detection accuracy respectively reaches 83.9% and 98.9% for the NEU-DET dataset and DeepPCB dataset, which are better than the state-of-the-art defect detection methods. The end-to-end fast and accurate detection for industrial surface defects is realized.
Keyword:
Surface defect detection
Feature enhance
Deep separable convolution
Dense join
Multi-scale feature weighted
YOLO
期刊
IF:
4.3
论文数:
1.6K
被引数:
3.6K
机构
引用论文
Thermal comfort, perceived air quality, and cognitive performance when personally controlled air movement is used by tropically acclimatized persons当热带适应的人使用个人控制的空气运动时，热舒适性，感知的空气质量和认知表现.png)
Indoor Air
OAAI
Indoor AirIF0
OAAI分享收藏Automatic Defect Detection of Fasteners on the Catenary Support Device Using Deep Convolutional Neural Network基于深度卷积神经网络的接触网支撑装置紧固件缺陷自动检测OAAI
OAAI分享收藏
查看更多内容