Lightweight YOLOv7 for bushing surface defects detection
摘要
Bushings have a wide range of applications in industry. Once the surface of the bushing is defective, it will affect the assembly between bearings resulting in mechanical inefficiency. At present, due to the different target sizes of the different types of defects on the bushing surface, it is difficult to balance inspection accuracy and speed. This paper proposes lightweight You Only Look Once (YOLO) v7 networks to cope with this problem. In this paper, we use a lightweight network, MobileNetv3, as the backbone network, in which a Residual edges CBAM block (RC-block) is designed to retain feature information while focusing on small-scale targets; finally, we use a bi-directional feature pyramid network (BiFPN) to perform feature fusion to further improve the detection accuracy. The experimental results show that the improved model reduces the Mean Average Precision (mAP) by only