Abstract:Surface defects in bearings not only affect the appearance but also impact the service life and performance. Therefore, it is imperative for bearing manufacturers to conduct quality inspections before bearings leave the factory. However, traditional visual inspection methods exhibit shortcomings such as high omission rates, insufficient feature fusion and oversized models when dealing with multiple target defects in bearings. To address these challenges, this paper proposes a surface defect detection method for bearings based on an improved Yolov8 algorithm (G-Yolov8). Firstly, a C3Ghost convolutional module based on the Ghost module is constructed in YOLOv8 to simplify model computational costs. Secondly, a Global Attention Mechanism (GAM) module is designed at the end of the backbone network to increase sensitivity to implicit small target area features and optimize feature extraction efficiency.Subsequently, a deep deformable convolution feature pyramid network (TDSFPN) is constructed by introducing the deformable convolutional networks version 2 (DCNv2) and the lightweight Content-Aware Reassembly of Features upsampling operator (CARAFE) to reduce sampling information loss and improve the fusion of multi-scale target defects. Finally, different attention mechanisms are embedded in the detection network to construct a Multi-Attention Detection Head (MADH) to replace the decoupled head, refining classification and localization tasks, reducing feature confusion, and improving the model's detection accuracy. Experimental results demonstrate that the improved algorithm achieves a 3.5% increase in mean average precision on a self-made small-scale train bearing surface defect dataset, with a 17.3% reduction in model size. This improvement not only enhances accuracy but also addresses the requirement for lightweight deployment in subsequent stages.

Small-modulus worms surface defect detection method based on YOLOv7

A Wafer Surface Defect Detection Method Built on Generic Object Detection Network

TBi-YOLOv5: A Surface Defect Detection Model for Crane Wire with Bottleneck Transformer and Small Target Detection Layer

Small-scale defect detection in industrial environment based on lightweight deep learning network

A Lightweight Detection Algorithm for Surface Defects in Small-Sized Bearings

A Small-scale Object Detection Method for LCD Defects Based on Improved YOLOv8

Optimized YOLOv8 based on SGW for surface defect detection of silicon wafer

A Novel YOLOv5_ES based on lightweight small object detection head for PCB surface defect detection

A Deep Learning-Based Surface Defect Inspection System Using Multiscale and Channel-Compressed Features

WLSD-YOLO: A Model for Detecting Surface Defects in Wood Lumber

A small sample nonstandard gear surface defect detection method

LCG-YOLO: A Real-Time Surface Defect Detection Method for Metal Components

Improved YOLOv7-based steel surface defect detection algorithm

GBCD-YOLO: A High-Precision and Real-Time Lightweight Model for Wood Defect Detection

ESD-YOLOv5: A Full-Surface Defect Detection Network for Bearing Collars

A real-time and efficient surface defect detection method based on YOLOv4

BiSPD-YOLO: Surface Defect Detection Method for Small Features and Low-resolution Images

A Lightweight Modified YOLOX Network Using Coordinate Attention Mechanism for PCB Surface Defect Detection

A Bearing surface defect detection method based on multi-attention mechanism Yolov8

A bolt defect detection method for transmission lines based on improved YOLOv5

Object Detection Algorithm for Surface Defects Based on a Novel YOLOv3 Model