Image Processing

Surface Defect Detection Method for Industrial Products Based on Photometric Stereo and Dual Stream Feature Fusion Network

  • HU Guanghua ,
  • TU Qianxi
Expand
  • School of Mechanical and Automotive Engineering,South China University of Technology,Guangzhou 510640,Guangdong,China

Received date: 2023-10-13

  Online published: 2024-03-04

Supported by

the Natural Science Foundation of Guangdong Province(2022A1515010806)

Abstract

Surface defect detection is an important part of the modern industrial production process. The existing visual defect detection methods generally achieve detection by analyzing a single RGB or grayscale image of the target object and using differential features between the defect and the background. They are suitable for objects with a large difference between the target and the background, such as the detection of metal surface oxidation and spot defects. However, the simple RGB image cannot effectively characterize the 3D defect features such as dents and bulges, which are mainly formed by depth changes, ultimately resulting in missed detection. To this end, this paper extracted the 3D geometric appearance information of the object surface to be tested according to multi-directional light imaging and photometric stereo principle. Next, the original multi-directional light images were effectively fused using the contrast pyramid fusion algorithm to obtain the enhanced 2D RGB fusion image features of the defects. Then, on the basis of the multi-target detection framework YOLOv5, with the above geometric appearance and RGB fusion images as inputs, a defect detection network model based on dual stream feature fusion detection network model was constructed. The model introduces the spatial channel attention residual module and the gated recurrent unit (GRU) feature fusion module and is able to organically fuse the different modal features at multiple levels to realize the effective extraction of the 2D RGB and 3D appearance information of the surface defects, so as to achieve the purpose of dealing with the detection of 2D and 3D defects at the same time. Finally, the detection experiments were conducted on the surface defects of several typical industrial products. The results show that mAP of the method in the paper is above 90% on several datasets, and it can simultaneously cope with the detection of 2D and 3D defects, so the detection performance is better than that of the current mainstream methods, and it can meet the detection requirements of different industrial products.

Cite this article

HU Guanghua , TU Qianxi . Surface Defect Detection Method for Industrial Products Based on Photometric Stereo and Dual Stream Feature Fusion Network[J]. Journal of South China University of Technology(Natural Science), 2024 , 52(10) : 112 -123 . DOI: 10.12141/j.issn.1000-565X.230638

References

1 WOODHAM R J .Photometric stereo:a reflectance map technique for determining surface orientation from image intensity[C]∥ Proceedings of SPIE, Image Understanding Systems and Industrial Applications Ⅰ.San Diego:SPIE, 1978:136-143.
2 SOUKUP D, HUBER-MORK R .Convolutional neural networks for steel surface defect detection from photometric stereo images[C]∥ Proceedings of the 10th International Symposium on Visual Computing.Las Vegas:Springer-Verlag,2014:668-677.
3 CAO Y, DING B, CHEN J,et al .Photometric-stereo-based defect detection system for metal parts[J].Sensors202222(21):8374/1-17.
4 REN S, HE K, GIRSHICK R,et al .Faster R-CNN:towards real-time object detection with region proposal networks[J].IEEE Transactions on Pattern Analysis and Machine Intelligence201739(6):1137-1149.
5 XIE R, ZHU Y, LUO J,et al .Detection algorithm for bearing roller end surface defects based on improved YOLOv5n and image fusion[J].Measurement Science and Technology202334(4):045402/1-11.
6 SAIZ F A, BARANDIARAN I, ARBELAIZ A,et al .Photometric stereo-based defect detection system for steel components manufacturing using a deep segmentation network[J].Sensors202222(3):882/1-14.
7 XIAO F, LI B, PENG Y,et al . Multi-modal weights sharing and hierarchical feature fusion for RGBD salient object detection[J]. IEEE Access20208:26602/1-11.
8 LI G, LIU Z, LING H .ICNet:information conversion network for RGB-D based salient object detection[J].IEEE Transactions on Image Processing202029:4873-4884.
9 LV H, DENG B, SHI C .Research on multi-source image fusion target detection technology based on neural network[J].Journal of Physics:Conference Series20212033:012139/1-5.
10 FU H, WANG S, DUAN P,et al .LRAF-Net:long-range attention fusion network for visible-infrared object detection[J].IEEE Transactions on Neural Networks and Learning Systems2023,(2023-06-06).Doi:10.1109/TNNLS.2023.3266452.
11 JIN X, GUO C, HE Z,et al . FCMNet:frequency-aware cross-modality attention networks for RGB-D salient object detection[J].Neurocomputing2022491:414-425.
12 金闳奇 .笔记本电脑铝合金面板表面缺陷检测系统 [D].广州:广东工业大学,2022
13 TREIBITZ T, SCHECHNER Y Y .Turbid scene enhancement using multi-directional illumination fusion [J].IEEE Transactions on Image Processing201221(11):4662-4667.
14 MNIH V, HEESS N, GRAVES A,et al . Recurrent models of visual attention[C]∥ Proceedings of the 28th Annual Conference on Neural Information Processing Systems.Montreal:Neural Information Processing Systems Foundation,2014:2204-2212.
15 WOO S, PARK J, LEE J-Y,et al .CBAM:convolutional block attention module[C]∥ Proceedings of the 15th European Conference on Computer Vision.Munich:Springer-Verlag,2018:3-19.
16 CHUNG J, GULCEHRE C, CHO K,et al .Empirical evaluation of gated recurrent neural networks on sequence modeling[EB/OL].(2014-11-11)[2023-10-12]. .
17 WANG W, XIE E, SONG X,et al .Efficient and accurate arbitrary-shaped text detection with pixel aggregation network[C]∥ Proceedings of 2019 IEEE/CVF International Conference on Computer Vision.Seoul:IEEE,2019:8439-8448.
18 YU J, JIANG Y, WANG Z,et al .UnitBox:an advanced object detection network[C]∥ Proceedings of the 24th ACM International Conference on Multimedia. Amsterdam:ACM,2016:516-520.
19 LIU W, ANGUELOV D, ERHAN D,et al .SSD:single shot multibox detector[C]∥ Proceedings of the 14th European Conference on Computer Vision.Amsterdam:Springer-Verlag,2016:21-37.
Outlines

/