[发明专利]一种基于多源特征的磁芯缺陷检测模型训练方法及系统

权利要求书

1.一种基于多源特征的磁芯缺陷检测模型训练方法，其特征在于，包括：

步骤S101，获取目标磁芯的自然光成像图片和红外线成像图片并作为训练集进行模型训练，得到当前轮磁芯缺陷检测模型；

所述模型训练包括：

步骤S1011，将所述自然光成像图片和所述红外线成像图片进行位置校正，得到校正后的自然光成像图片和校正后的红外线成像图片；

步骤S1012，将所述校正后的自然光成像图片和校正后的红外线成像图片分别分成第一预设数量块，并一一对应进行融合，得到第一预设数量个融合特征块；

步骤S1013，将所述第一预设数量个融合特征块进行拼接，得到多源特征图；将所述多源特征图依次进行分块操作、自注意力机制操作、全卷积操作、卷积操作、融合操作得到总自注意力特征图；将所述总自注意力特征图输入到所述模型的检测头中进行检测，得到特征图中磁芯缺陷的位置以及所述当前轮磁芯缺陷检测模型；根据所述特征图中磁芯缺陷的位置与所述校正后的自然光成像图片或校正后的红外线成像图片中磁芯缺陷的位置计算损失值；

步骤S102，重复步骤S101直至损失值在预设范围内波动，得到目标磁芯缺陷检测模型；

其中，所述步骤S1013包括：

将所述第一预设数量个融合特征块进行拼接，得到多源特征图；

将所述多源特征图划分为第二预设数量块，并将每一块进行自注意力机制操作，得到第二预设数量个自注意力特征图；将所述第二预设数量个自注意力特征图进行融合并进行全卷积操作得到第一块自注意力特征图；

将所述第一块自注意力特征图划分为第三预设数量块，并将每一块进行自注意力机制操作，得到第三预设数量个自注意力特征图；将所述第三预设数量个自注意力特征图进行融合并进行全卷积操作得到第二块自注意力特征图；

将所述第二块自注意力特征图划分为第四预设数量块，并将每一块进行自注意力机制操作，得到第四预设数量个自注意力特征图；将所述第四预设数量个自注意力特征图进行融合并进行全卷积操作得到第三块自注意力特征图；

将所述第一块自注意力特征图、所述第二块自注意力特征图、所述第三块自注意力特征图依次进行卷积操作、融合操作得到总自注意力特征图；将所述总自注意力特征图输入到所述模型的检测头中进行检测，得到特征图中磁芯缺陷的位置以及所述当前轮磁芯缺陷检测模型；根据所述特征图中磁芯缺陷的位置与所述校正后的自然光成像图片或校正后的红外线成像图片中磁芯缺陷的位置计算损失值。

2.根据权利要求1所述的方法，其特征在于，所述步骤S1012包括：

将所述校正后的自然光成像图片和校正后的红外线成像图片分别分成第一预设数量块，并一一对应按通道连接，得到第一预设数量个通道叠加图片块；

将所述第一预设数量个通道叠加图片块进行自注意力机制操作，得到第一预设数量个特征图中每个特征值的相关性；其中，所述相关性为特征图中一个位置的特征值与其他位置的特征值的关系；

根据所述第一预设数量个特征图中每个特征值的相关性进行卷积操作，以使所述第一预设数量个通道叠加图片块的通道与所述校正后的自然光成像图片或校正后的红外线成像图片的通道相等，得到第一预设数量个融合特征块。

3.根据权利要求2所述的方法，其特征在于：

所述第一预设数量个融合特征块的计算公式为：

其中，所述Feature为一个融合特征块；所述Q×K^T为一个特征图中每个特征值的相关性，所述相关性为特征图中一个位置的特征值与其他位置的特征值的关系；为一个特征图上的每个特征值的位置与它关系最密切的特征值位置；为一个特征图；Conv为卷积操作，以使一个通道叠加块的通道与所述校正后的自然光成像图片和红外线成像图片的通道相等。