Improved RetinaNet-Based Defect Detection for Engine Parts

Zhipei Qi,Mengyi Zhang,Jun Li,Cuimei Bo,Cunsong Wang,Hao Peng
DOI: https://doi.org/10.23919/CCC58697.2023.10241037
2023-01-01
Abstract:The safety of aviation, vehicles, and ships relies heavily on the quality assurance of engine parts. However, traditional target detection algorithms currently used to detect defects in engine parts do not have a high recognition rate. This paper proposes an improved RetinaNet to address this problem. Firstly, the backbone network ResNet was improved by replacing <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$7\times 7$</tex> convolutional kernels with stacked <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$3\times 3$</tex> convolutional kernels, and the C2 layer of ResNet was densely connected to C3, C4, and C5 layers to improve the feature extraction capability. Then we improved the feature pyramid (FPN) structure by adding a convolutional attention module (CBAM) after layers P3 to P5 of the feature pyramid. Finally, the improved RetinaNet was trained on a self-made engine part missing detection dataset, and the model performance was tested separately. The experimental results show that RetinaNet has better performance in detecting engine part defects, but there is a small part of missed detection. The improved model performs well in detecting small targets, and the mean average precision (mAP) reaches 88.4%, which is 1.8% higher than the mAP of the original RetinaNet, furthermore, the average detection time is 0.22s. In the practical scene application, it can quickly and accurately detect defects in engine parts.
What problem does this paper attempt to address?