Surface Defect Detection Model for Aero-Engine Components Based on Improved YOLOv5
Source abstract
Abstract
Aiming at the problems of low efficiency and poor accuracy in conventional surface defect detection methods for aero-engine components, a surface defect detection model based on an improved YOLOv5 object detection algorithm is proposed in this paper. First, a k-means clustering algorithm was used to recalculate the parameters of the preset anchors to make them match the samples better. Then, an ECA-Net attention mechanism was added at the end of the backbone network to make the model pay more attention to feature extraction from defect areas. Finally, the PANet structure of the neck network was improved through its replacement with BiFPN modules to fully integrate the features of all scales. The results showed that the mAP of the YOLOv5s-KEB model was 98.3%, which was 1.0% higher than the original YOLOv5s model, and the average inference time for a single image was 2.6 ms, which was 10.3% lower than the original model. Moreover, compared with the Faster R-CNN, YOLOv3, YOLOv4 and YOLOv4-tiny object detection algorithms, the YOLOv5s-KEB model has the highest accuracy and the smallest size, which make it very efficient and convenient for practical applications.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Industrial Vision Systems and Defect Detection · Advanced Neural Network Applications · 3D Surveying and Cultural Heritage
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Fundamentals Department, Air Force Engineering University, Xi’an 710051, China
Funding
National Natural Science Foundation of China
References (28)
Show source-provided references
1. Deep learning-based borescope image processing for aero-engine blade in-situ damage detection10.1016/j.ast.2022.107473 ↗
2. A repair and overhaul methodology for aeroengine components10.1016/j.rcim.2009.07.001 ↗
3. Tiny defect detection in high-resolution aero-engine blade images via a coarse-to-fine framework
4. Girshick, R., Donahue, J., Darrell, T., and Malik, J. (2014, January 23–28). Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Columbus, OH, USA.10.1109/CVPR.2014.81 ↗
5. Girshick, R. (2015, January 7–13). Fast R-CNN. Proceedings of the IEEE International Conference on Computer Vision, Santiago, Chile.10.1109/ICCV.2015.169 ↗
6. FasterR-CNN: Towards Real-time Object Detection with Region Proposal Networks
7. Redmon, J., Divvala, S., and Girshick, R. (2016, January 12). You Only Look Once: Unified, Real-Time Object Detection. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA.10.1109/CVPR.2016.91 ↗
8. Liu, W., Anguelov, D., Erhan, D., Szegedy, C., Reed, S., Fu, C.Y., and Berg, A.C. (2016, January 11–14). SSD: Single Shot Multibox Detector. Proceedings of the European Conference on Computer Vision, Amsterdam, The Netherlands.10.1007/978-3-319-46448-0_2 ↗
9. Do vision transformers see like convolutional neural networks?
10. Detection of objects in the images: From likelihood relationships towards scalable and efficient neural networks10.18287/2412-6179-CO-922 ↗
11. Ramchandran, A., and Sangaiah, A.K. (2018). Unsupervised anomaly detection for high dimensional data—An exploratory analysis. Computational Intelligence for Multimedia Big Data on the Cloud with Engineering Applications, Academic Press.10.1016/B978-0-12-813314-9.00011-6 ↗
12. Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., and Gelly, S. (2020). An image is worth 16 × 16 words: Transformers for image recognition at scale. arXiv.
13. Development of a YOLO-V3-based model for detecting defects on steel strip surface10.1016/j.measurement.2021.109454 ↗
14. Andriyanov, N., Khasanshin, I., Utkin, D., Gataullin, T., Ignar, S., Shumaev, V., and Soloviev, V. (2022). Intelligent system for estimation of the spatial position of apples based on YOLOv3 and real sense depth camera D415. Symmetry, 14.10.3390/sym14010148 ↗
15. A review on modern defect detection models using DCNNs–Deep convolutional neural networks10.1016/j.jare.2021.03.015 ↗
16. Du, X., Chen, J., Zhang, H., and Wang, J. (2022). Fault detection of aero-engine sensor based on inception-CNN. Aerospace, 9.10.3390/aerospace9050236 ↗
17. Redmon, J., and Farhadi, A. (2017, January 21–26). YOLO9000: Better, Faster, Stronger. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA.10.1109/CVPR.2017.690 ↗
18. Redmon, J., and Farhadi, A. (2018). Yolov3: An incremental improvement. arXiv.
19. Bochkovskiy, A., Wang, C.Y., and Liao, H.Y.M. (2020). Yolov4: Optimal speed and accuracy of object detection. arXiv.
20. Liu, S., Qi, L., Qin, H., Shi, J., and Jia, J. (2018, January 18–23). Path Aggregation Network for Instance Segmentation. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA.10.1109/CVPR.2018.00913 ↗
21. Lin, T., Dollar, P., Girshick, R., He, K., Hariharan, B., and Belongie, S. (2017, January 21–26). Feature Pyramid Networks for Object Detection. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA.10.1109/CVPR.2017.106 ↗
22. GitHub (2021, March 01). YOLOv5-Master. Available online: https://github.com/ultralytics/yolov5.git/.
23. Wang, C.-Y., Mark Liao, H.-Y., Wu, Y.-H., Chen, P.-Y., Hsieh, J.-W., and Yeh, I.-H. (2020, January 13–19). CSPNet: A New Backbone that can Enhance Learning Capability of CNN. Proceedings of the Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.10.1109/CVPRW50498.2020.00203 ↗
24. Wang, Q., Wu, B., Zhu, P., Li, P., Zuo, W., and Hu, Q. (2020, January 13–19). ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks. Proceedings of the Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.10.1109/CVPR42600.2020.01155 ↗
25. Squeeze-and-Excitation networks10.1109/TPAMI.2019.2913372 ↗
26. Improved edge lightweight YOLOv4 and its application in on-site power system work10.1016/j.gloei.2022.04.014 ↗
27. Tan, M., Pang, R., and Le, Q.V. (2020, January 13–19). EfficientDet: Scalable and Efficient Object Detection. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.10.1109/CVPR42600.2020.01079 ↗
28. Du, F.J., and Jiao, S.J. (2022). Improvement of lightweight convolutional neural network model based on YOLO algorithm and its research in pavement defect detection. Sensors, 22.10.3390/s22093537 ↗