期刊
COMPUTERS & ELECTRICAL ENGINEERING
卷 102, 期 -, 页码 -出版社
PERGAMON-ELSEVIER SCIENCE LTD
DOI: 10.1016/j.compeleceng.2022.108237
关键词
Ghost module; Lightweight; YOLO; Damage detection; Deep learning
类别
资金
- Guangdong Province Key Areas R D Program [2019B010150002]
- Shanxi Provincial Philosophy and Social Science Planning Project [2021YY198]
- Shanxi Datong University Scientific Research YunGang Special Project [2020YGZX014, 2021YGZX27]
This paper proposes a lightweight neural network-based algorithm for non-contact damage detection of grotto murals, which effectively detects minor defects and improves detection performance by reducing network size and introducing attention mechanism.
Non-contact damage detection of grotto murals is a challenging task, due to their minor cracks and subtle defects. In this paper, a lightweight neural network-based grotto mural damage detection algorithm, named Ghost-C3SE YOLOv5, is proposed, which can effectively reduce the network size while obtaining reasonable damage detection results. First, based on the YOLOv5 detection algorithm, the dimension of the convolution layer is reduced, by adjusting the network structure and integrating a Ghost module. Second, the channel attention mechanism is introduced into the feature extraction backbone network, to adjust the weight of each feature according to its importance, which can accelerate the convergence speed of the loss function during the model training process. Finally, the experiment results have shown that the lightweight model Ghost-C3SE YOLOv5, applied on the Pascal Voc dataset, reduces the number of parameters by about 22.55 million while ensuring the detection precision and recall rate. Also, the training time and the model size are reduced by 36.21% and 46.04%, respectively. Precision is increased by 1.29% and recall remains comparative, while the utilization rate of the GPU is improved by 41.55%. This has addressed the shortcomings of laborious process and low accuracy in manual detection of the grotto murals. Furthermore, a real-time detection performance of 44.86 FPS is achieved here for damage detection of murals in the YunGang Grottoes, which has overcome the drawbacks of high model complexity, high computational cost, slow detection speed and low memory utilization of various existing deep learning algorithms.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据