Metal surface defect detection based on grouped convolution and bidirectional feature fusion
AIIPCC 2022; The Third International Conference on Artificial Intelligence, Information Processing and Cloud Computing(2022)
摘要
In this paper, we propose a metal surface defect detection model based on grouped convolutions and bidirectional feature fusion to solve the problems of dense distribution, diverse types, and small contrast between object and background. The model uses a multi-branch structure based on grouped convolutions from ResNet residual blocks of the feature extraction network layer, which increases the cardinality and feature extraction ability of the network. In feature fusion, we introduce a path enhancement network and a bottom-up path in the network to form a bidirectional feature fusion module, which fuses information in shallower layers and enhances the detection ability for small objects. We add a regression scale to reduce the amount of computation on the ordinary small objects’ regression layer. The convolution layer is shared between the center-ness branch and the regression branch to better suppress prediction boxes far away from the object center. Experimental results on NEU-DET metal surface defect dataset show that the average accuracy of our network model reaches 90.2%, which is 8.3% higher than that of Fully Convolutional One Stage (FCOS), thus our model can obviously improve the detection rate of metal surface defects.
更多查看译文
AI 理解论文
溯源树
样例
![](https://originalfileserver.aminer.cn/sys/aminer/pubs/mrt_preview.jpeg)
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要