[发明专利]一种太赫兹图像无损检测的方法、装置以及设备

说明书

本发明公开了一种太赫兹图像无损检测的方法、装置、设备及计算机可读存储介质，包括：将待检测的太赫兹图像输入至深度卷积神经网络，提取特征图；利用注意力机制、太赫兹图像的数据属性、第一目标关系矩阵和第一目标编码矩阵，将特征图转换为注意图；利用注意力机制、所述数据属性、第二目标关系矩阵和第二目标编码矩阵，将注意图转换为目标图像特征；将目标图像特征转换为规范共同特征；利用规范共同特征、目标分类权值矩阵和Softmax函数，确定太赫兹图像的损伤分类结果，从而确定太赫兹图像的损伤类别和损伤程度。本发明所提供的方法、装置、设备及计算机可读存储介质，利用多重注意力机制，准确表示太赫兹图像的损伤特征。

技术领域

本发明涉及图像处理技术领域，特别是涉及一种太赫兹图像无损检测的方法、装置、设备以及计算机可读存储介质。

背景技术

近年来，国内外学者利用太赫兹成像技术在对材料的无损探伤检测方面进行了相关的研究，特别是在探伤成像的后期图像增强方面进行了一定的探索研究。目前太赫兹图像无损检测的技术方法主要包括：基于小波变换与利普系茨指数的太赫兹图像的无损检测和基于小波去噪法的无损检测的方法。基于小波变换与利普系茨指数的太赫兹图像的无损检测丢失了损伤属性信息，对损伤特征表示十分模糊，最后的分类结果需要人工辅助加以实现。基于小波去噪法的无损检测的方法，一定程度上提高了太赫兹图像的分辨率，但也提升了损伤特征的冗余性，容易引起对损伤类别及程度的误判。

综上所述可以看出，如何准确表示太赫兹图像损伤特征是目前有待解决的问题。

发明内容

本发明的目的是提供一种太赫兹图像无损检测的方法、装置、设备以及计算机可读存储介质，以解决现有技术所提供的无损检测方法不能准确表示太赫兹图像损伤特征的问题。

为解决上述技术问题，本发明提供一种太赫兹图像无损检测的方法，包括：将待检测的太赫兹图像输入至深度卷积神经网络，提取所述太赫兹图像的特征图；利用注意力机制、所述太赫兹图像的数据属性、预先完成训练的第一目标关系矩阵和预先完成训练的第一目标编码矩阵，将所述特征图转换为注意图；利用所述注意力机制、所述数据属性、预先完成训练的第二目标关系矩阵和预先完成训练的第二目标编码矩阵，将所述注意图转换为目标图像特征；将所述目标图像特征转换为规范共同特征；利用规范共同特征、预先完成训练的目标分类权值矩阵和Softmax函数，确定所述太赫兹图像的损伤分类结果，从而确定所述太赫兹图像的损伤类别和损伤程度。

优选地，所述利用注意力机制、所述太赫兹图像的数据属性、预先完成训练的目标第一关系矩阵和预先完成训练的目标第一编码矩阵，将所述特征图转换为注意图包括：

根据所述数据属性、所述第一目标关系矩阵和所述特征图的乘积，确定所述数据属性在所述特征图的第一嵌入矩阵；

利用所述第一目标编码矩阵和所述特征图的乘积，确定所述特征图编码后的第一特征；

根据所述第一嵌入矩阵和所述第一特征的乘积，确定所述注意图。

优选地，所述利用所述注意力机制、所述数据属性、预先完成训练的目标第二关系矩阵和预先完成训练的目标第二编码矩阵，将所述注意图转换为目标图像特征包括：

根据所述数据属性、所述第二目标关系矩阵和所述注意图的乘积，确定所述数据属性在所述注意图的第二嵌入矩阵；

利用所述第二目标编码矩阵和所述注意图的乘积，确定所述注意图编码后的第二特征；

根据所述第二嵌入矩阵和所述第二特征的乘积，确定所述目标图像特征。

优选地，还包括：

将已知类别标签的太赫兹图像训练集中每幅初始太赫兹图像分别输入至所述深度卷积神经网络中，提取所述每幅初始太赫兹图像的初始特征图；