Abstract
In the detection of defects on the surface of aluminum materials, such as jet stream and scratch. These defects have the problem of background similarity of different sizes, which brings difficulties and challenges to the detection. Our method introduces a cross-layer linking network to further fuse shallow and deep features. Taking into account more location, semantics, and detailed information, the detection accuracy of the network for aluminum surface defects is improved. Moreover, A weighted feature fusion mechanism is introduced to solve the negative impact of features from different layers and enhance the feature extraction ability of the model. Experimental results show that the improved YOLOv5 network model has good defect detection performance. The mAP on the Tianchi dataset reached 78.4%, which is 2.3% higher than the original YOLOv5 network. The model in this paper can quickly and accurately detect aluminum surface defects while keeping the original detection speed basically unchanged.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Mao, T.Q., et al.: Defect recognition method based on HOG and SVM for drone inspection images of power transmission line. In: 2019 International Conference on High Performance Big Data and Intelligent Systems, pp. 254–257. IEEE Press, New York (2019)
Chu, M.X., et al.: Steel surface defects recognition based on multi-type statistical features and enhanced twin support vector machine. Chemom. Intell. Lab. Syst. 171, 140–150 (2017)
Ren, S.Q., et al.: Faster R-CNN: towards real-time object detection with region proposal networks. IEEE Trans. Pattern Anal. Mach. Intell. 39(6), 1137–1149 (2017)
Redmon, J., et al.: You only look once: unified, real-time object detection. In: 2016 IEEE Conference on Computer Vision and Pattern Recognition, pp. 779–788. IEEE Press, New York (2016)
Chang, L., Chen, Y.-T., Hung, M.-H., Wang, J.-H., Chang, Y.-L.:YOLOV3 based ship detection in visible and infrared images. In: 2021 IEEE International Geoscience and Remote Sensing Symposium, IGARSS, Brussels, Belgium, pp. 3549–3552 (2021)
Song, D., Yuan, F., Ding, C.: Track foreign object debris detection based on improved YOLOv4 model. In: 2022 IEEE 6th Advanced Information Technology, Electronic and Automation Control Conference (IAEAC), Beijing, China, pp. 1991–1995 (2022)
Dong, Z., Chang, H., Pu, X., Luo, P., Weng, J., Liu, Z.:Aircraft segmentation algorithm based on Unet and improved Yolov4. In: 2021 International Conference on Intelligent Transportation, Big Data & Smart City (ICITBS), Xi’an, China, pp. 614–617 (2021)
Huang, M., Wang, B., Wan, J., Zhou, C.:Improved blood cell detection method based on YOLOv5 algorithm. In: 2023 IEEE 6th Information Technology, Networking, Electronic and Automation Control Conference (ITNEC), Chongqing, China, pp. 992–996 (2023)
Li, Y., Tang, B., Sun, W., Lin, Z., Li, J.: Detection of steel plate surface defects based on improved faster R-CNN. Modular Mach. Tool Autom. Manuf. Tech. (05), 113–115+119 (2022)
Li Yanzhao, Y., Jinwen, L.T.: Metal weld defect detection based on improved YOLOv5. Electr. Meas. Technol. 45(19), 70–75 (2022)
Zhao-ming, G.E., Yue-ming, H.U.: Lithium battery electrode defect detection method based on improved YOLOv5. Laser J. 44(02), 25–29 (2023)
Xiaomeng, L., Jun, F., Peng, C.:Vehicle detection in traffic monitoring scenes based on improved YOLOV5s. In: 2022 International Conference on Computer Engineering and Artificial Intelligence (ICCEAI), Shijiazhuang, China, pp. 467–471 (2022)
Wang, X., Zhang, X., Zhou, N.:Improved YOLOv5 with BiFPN on PCB defect detection. In: 2021 2nd International Conference on Artificial Intelligence and Computer Engineering (ICAICE), Hangzhou, China, pp. 196–199 (2021)
Liu, W., Quijano, K., Crawford, M.M.: YOLOv5-tassel: detecting tassels in RGB UAV imagery with improved YOLOv5 based on transfer learning. IEEE J. Sel. Top. Appl. Earth Observations Remote Sens. 15, 8085–8094 (2022)
Acknowledgement
This work was supported by National Natural Science Foundation of China under Grant No. 62103256.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 ICST Institute for Computer Sciences, Social Informatics and Telecommunications Engineering
About this paper
Cite this paper
Sui, T. et al. (2024). Aluminum Defect Detection Based on Weighted Feature Fusion Mechanism. In: Li, J., Zhang, B., Ying, Y. (eds) 6GN for Future Wireless Networks. 6GN 2023. Lecture Notes of the Institute for Computer Sciences, Social Informatics and Telecommunications Engineering, vol 553. Springer, Cham. https://doi.org/10.1007/978-3-031-53401-0_17
Download citation
DOI: https://doi.org/10.1007/978-3-031-53401-0_17
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-53400-3
Online ISBN: 978-3-031-53401-0
eBook Packages: Computer ScienceComputer Science (R0)