基于轻量化神经网络的石窟壁画破损检测方法

Damage Detection Method for Grotto Murals Based on Lightweight Neural Network

  • 摘要: 针对石窟壁画脱落与破损检测过程中存在检测精度低、实时性差的问题,提出了基于轻量化神经网络和多重注意力机制的石窟壁画破损检测方法。首先,引入Ghost Conv完成轻量化特征提取,降低模型复杂度;其次,加入双重注意力机制增加特征提取的倾向性,加快模型收敛速度;最后,使用加权双向特征金字塔拼接方式高效融合特征信息,通过复合缩放完成预测。实验结果表明:改进后的算法网络层数减少了34.40%。参数量和浮点运算量分别降低了62.98%和68.77%,模型体积压缩了62.78%。检测精度高达64.7%,实时检测速度从63.60帧/s提升至97.56帧/s,提高了约53.39%。

     

    Abstract: To address the issues of low detection precision and poor real-time performance in the process of grotto mural detachment and damage detection, we propose a grotto mural damage detection method based on a lightweight neural network and multiple attention mechanisms. First, Ghost Conv is introduced to complete lightweight feature extraction and reduce model complexity. Second, we add a double attention mechanism to increase the tendency of feature extraction and accelerate model convergence. Finally, we use a weighted bidirectional feature pyramid network to efficiently fuse feature information and complete prediction by composite scaling. The experimental results show that the improved algorithm reduces the number of network layers by 34.40%. The number of parameters and floating point operations are reduced by 62.98% and 68.77%, respectively, and the model volume is compressed by 62.78%. The detection precision is 64.7%, and the real-time detection speed is improved from 63.60 frame/s to 97.56 frame/s, which is approximately 53.39%.

     

/

返回文章
返回