The invention provides a lightweight
bearing surface defect
small target detection method based on improved YOLOv8. According to the method, based on a YOLOv8n
network architecture, a dynamic detection head is constructed, an ADown module is introduced, a
small target detection layer is additionally arranged, and finally an improved YOLOv8n-DAS model is established; the method comprises the following steps: firstly, constructing a dynamic detection head, fusing scale, space and task
perception attention mechanisms, intensifying feature expression ability in all directions, and accurately capturing complex tiny defect features on the surface of a bearing; secondly, the ADown module highlights edge defect information, simplifies the
model parameter scale and reduces consumption of computing resources by combining traditional
convolution, average
pooling and maximum
pooling operations; finally,
small target detection
layers are arranged at the neck and the head of the network, the small target detection
layers are added, an extra small target detection head is arranged, shallow details and deep
semantic information are deeply integrated, key details are reserved to the maximum extent, and the recognition capacity of small targets is improved. According to the method, the
detection performance of the model on irregular and tiny defects and the detection capability of the model on small target defects are remarkably improved, the parameter quantity and the calculation complexity of the model are reduced, and deployment and application on lightweight equipment with
limited resources are facilitated. And the method can be expanded to defect detection of industrial parts such as gears and blades by replacing training data, has the characteristics of light weight and high universality, and meets multi-scene deployment requirements.