Modern Manufacturing Engineering ›› 2024, Vol. 531 ›› Issue (12): 120-129.doi: 10.16731/j.cnki.1671-3133.2024.12.016

Previous Articles     Next Articles

A small target defect detection method for aluminum profile surface based on improved YOLOv8n

SUN Tieqiang, LIU Jun, SONG Chao, XIAO Pengcheng   

  1. College of Artificial Intelligence,North China University of Science and Technology, Tangshan 063210,China
  • Received:2024-05-23 Online:2024-12-18 Published:2024-12-24

Abstract: To address the issues of slow surface defect detection speed in industrial aluminum profiles,as well as low accuracy in detecting small target defects such as scratches and paint bubbles,a small target aluminum profile surface defect detection model based on improved YOLOv8n was proposed. Firstly,cross-scale feature fusion was employed to enhance the network's ability to perceive multiscale information,by integrating features from different scales. Secondly,the C2f structure of the YOLOv8 backbone network was improved by incorporating RFAConv attention convolution. This was done to design a new structure to enhance the network's feature extraction ability and improve the accuracy of small object detection. In the neck region,lightweight RepGhost modules were used to reduce parameter count and computational complexity. Finally,the original YOLOv8's CIoU loss function was replaced with WIoUV3 to improve detection accuracy and convergence speed.Extensive experiments on aluminum profile defect datasets showed the model's effectiveness. It achieved a 2.5 % increase in mAP@0.5,The frame rate has reached 110 fps,and reduced weight file size by 16.2 %,surpassing the original model's performance. Experiments on the VOC2012 dataset and the Northeastern University hot rolled strip surface defect dataset demonstrated the algorithm's robustness.The refined model effectively meets industrial demands for aluminum profile detection.

Key words: surface defect detection, attention convolution, feature fusion, YOLOv8, small object detection

CLC Number: 

Copyright © Modern Manufacturing Engineering, All Rights Reserved.
Tel: 010-67126028 E-mail: 2645173083@qq.com
Powered by Beijing Magtech Co. Ltd