一种结合注意力机制的3D-CNN探地雷达三维图像管线识别方法

    公开(公告)号:CN115291210B

    公开(公告)日:2024-04-30

    申请号:CN202210881070.7

    申请日:2022-07-26

    Abstract: 本发明提出一种结合注意力机制的3D‑CNN探地雷达三维图像管线识别方法。所述方法包括对所获得的探地雷达的实际三维回波图像进行预处理;对得到的探地雷达的三维回波图像进行人工标注,并进行置乱,然后随机分配至训练集和验证集,但是需要保证每次置乱后的训练集中每个类别的样本数保持相同;利用生成的训练集对添加注意力机制的3D‑CNN的神经网络模型进行训练,获得训练好的神经网络模型;利用得到的神经网络模型,对探地雷达三维图像进行管线识别。本发明解决了传统识别方法以及基于2D‑CNN的神经网络识别方法识别准确率低和效率低的问题,同时通过添加注意力机制对3D‑CNN进行了改进,提高了网络模型对探地雷达三维图像的识别准确率。

Patent Agency Ranking