卷积神经网络
计算机科学
块(置换群论)
特征(语言学)
曲面(拓扑)
模式识别(心理学)
人工智能
特征提取
聚类分析
数学
几何学
语言学
哲学
作者
Xiancong Shi,Sike Zhou,Yichun Tai,Jinzhong Wang,Shoucang Wu,Jinrong Liu,Kun Xu,Tao Peng,Zhijiang Zhang
标识
DOI:10.1109/mmsp55362.2022.9949350
摘要
With the continuous increase of surface quality requirements in steel industry manufacturing, defect detection has received extensive attention. Correct and rapid detection of steel surface defects can significantly improve product quality and productivity. The existing methods improve accuracy by expanding the depth of networks or using various feature fusion technologies, but reduce the computational efficiency. To achieve the balance of precision and speed, we propose an improved network based on Faster R-CNN for defect detection of steel surface. Firstly, the emerging ConvNeXt architecture is adopted to act as backbone to extract features in Faster R-CNN. Besides, Convolutional Block Attention Module (CBAM) is used to improve the attention of our model to surface defects and suppress features of the complex background. Finally, k-means clustering algorithm is utilized to generate anchors that are better adapted to the surface defects. The proposed method achieves a mean average precision (mAP) of 80.78% and a detection speed of 26 frames per second (FPS) on the NEU-DET dataset, improving about 1.5% compared to the YOLOv5 and 8.4 % compared to original Faster R-CNN, which indicates its superior for defect detection of steel surface.
科研通智能强力驱动
Strongly Powered by AbleSci AI