技术特征：
1.一种基于多源特征的磁芯缺陷检测模型训练方法，其特征在于，包括：步骤s101，获取目标磁芯的自然光成像图片和红外线成像图片并作为训练集进行模型训练，得到当前轮磁芯缺陷检测模型；所述模型训练包括：步骤s1011，将所述自然光成像图片和所述红外线成像图片进行位置校正，得到校正后的自然光成像图片和校正后的红外线成像图片；步骤s1012，将所述校正后的自然光成像图片和校正后的红外线成像图片分别分成第一预设数量块，并一一对应进行融合，得到第一预设数量个融合特征块；步骤s1013，将所述第一预设数量个融合特征块进行拼接，得到多源特征图；将所述多源特征图依次进行分块操作、自注意力机制操作、全卷积操作、卷积操作、融合操作得到总自注意力特征图；将所述总自注意力特征图输入到所述模型的检测头中进行检测，得到特征图中磁芯缺陷的位置以及所述当前轮磁芯缺陷检测模型；根据所述特征图中磁芯缺陷的位置与所述校正后的自然光成像图片或校正后的红外线成像图片中磁芯缺陷的位置计算损失值；步骤s102，重复步骤s101直至损失值在预设范围内波动，得到目标磁芯缺陷检测模型。2.根据权利要求1所述的方法，其特征在于，所述步骤s1012包括：将所述校正后的自然光成像图片和校正后的红外线成像图片分别分成第一预设数量块，并一一对应按通道连接，得到第一预设数量个通道叠加图片块；将所述第一预设数量个通道叠加图片块进行自注意力机制操作，得到第一预设数量个特征图中每个特征值的相关性；其中，所述相关性为特征图中一个位置的特征值与其他位置的特征值的关系；根据所述第一预设数量个特征图中每个特征值的相关性进行卷积操作，以使所述第一预设数量个通道叠加图片块的通道与所述校正后的自然光成像图片或校正后的红外线成像图片的通道相等，得到第一预设数量个融合特征块。3.根据权利要求2所述的方法，其特征在于：所述第一预设数量个融合特征块的计算公式为：其中，所述为一个融合特征块；所述为一个特征图中每个特征值的相关性，所述相关性为特征图中一个位置的特征值与其他位置的特征值的关系；为一个特征图上的每个特征值的位置与它关系最密切的特征值位置；为一个特征图；为卷积操作，以使一个通道叠加块的通道与所述校正后的自然光成像图片和红外线成像图片的通道相等。4.根据权利要求1所述的方法，其特征在于，所述步骤s1013包括：将所述第一预设数量个融合特征块进行拼接，得到多源特征图；
将所述多源特征图划分为第二预设数量块，并将每一块进行自注意力机制操作，得到第二预设数量个自注意力特征图；将所述第二预设数量个自注意力特征图进行融合并进行全卷积操作得到第一块自注意力特征图；将所述第一块自注意力特征图划分为第三预设数量块，并将每一块进行自注意力机制操作，得到第三预设数量个自注意力特征图；将所述第三预设数量个自注意力特征图进行融合并进行全卷积操作得到第二块自注意力特征图；将所述第二块自注意力特征图划分为第四预设数量块，并将每一块进行自注意力机制操作，得到第四预设数量个自注意力特征图；将所述第四预设数量个自注意力特征图进行融合并进行全卷积操作得到第三块自注意力特征图；将所述第一块自注意力特征图、所述第二块自注意力特征图、所述第三块自注意力特征图依次进行卷积操作、融合操作得到总自注意力特征图；将所述总自注意力特征图输入到所述模型的检测头中进行检测，得到特征图中磁芯缺陷的位置以及所述当前轮磁芯缺陷检测模型；根据所述特征图中磁芯缺陷的位置与所述校正后的自然光成像图片或校正后的红外线成像图片中磁芯缺陷的位置计算损失值。5.根据权利要求4所述的方法，其特征在于，所述将所述第一块自注意力特征图、所述第二块自注意力特征图、所述第三块自注意力特征图依次进行卷积操作、融合操作得到总自注意力特征图包括：将所述第一块自注意力特征图、所述第二块自注意力特征图、所述第三块自注意力特征图进行卷积操作，以使所述第一块自注意力特征图、所述第二块自注意力特征图、所述第三块自注意力特征图的大小尺度相等；将大小尺度相等的所述第一块自注意力特征图、所述第二块自注意力特征图、所述第三块自注意力特征图进行融合得到总自注意力特征图。6.根据权利要求1所述的方法，其特征在于，所述步骤s1011包括：通过自然目标框对所述自然光成像图片的缺陷进行标定；以及，通过红外目标框对所述红外线成像图片中的缺陷进行标定；将所述自然目标框和所述红外目标框进行位置校正；以及，将所述自然光成像图片和所述红外线成像图片进行位置校正，得到校正后的自然光成像图片和校正后的红外线成像图片。7.根据权利要求6所述的方法，其特征在于，所述将所述自然目标框和所述红外目标框进行位置校正包括：将所述自然目标框的宽与高的倾斜角、所述红外目标框的宽与高的倾斜角进行校正，以使两个倾斜角相等；将所述自然目标框的宽与高的长度、所述红外目标框的宽与高的长度进行校正，以使两个宽相等、两个长相等；将所述自然目标框的四个坐标和所述红外目标框的四个坐标进行校正，以使两个目标框的四个坐标相等。8.一种基于多源特征的磁芯缺陷检测模型训练系统，其特征在于，包括：当前轮训练单元，用于获取目标磁芯的自然光成像图片和红外线成像图片并作为训练集进行模型训练，得到当前轮磁芯缺陷检测模型；
所述模型训练包括：校正子单元，用于将所述自然光成像图片和所述红外线成像图片进行位置校正，得到校正后的自然光成像图片和校正后的红外线成像图片；融合子单元，用于将所述校正后的自然光成像图片和校正后的红外线成像图片分别分成第一预设数量块，并一一对应进行融合，得到第一预设数量个融合特征块；检测子单元，用于将所述第一预设数量个融合特征块进行拼接，得到多源特征图；将所述多源特征图依次进行分块操作、自注意力机制操作、全卷积操作、卷积操作、融合操作得到总自注意力特征图；将所述总自注意力特征图输入到所述模型的检测头中进行检测，得到特征图中磁芯缺陷的位置以及所述当前轮磁芯缺陷检测模型；根据所述特征图中磁芯缺陷的位置与所述校正后的自然光成像图片或校正后的红外线成像图片中磁芯缺陷的位置计算损失值；重复训练单元，重复所述当前轮训练单元直至损失值在预设范围内波动，得到目标磁芯缺陷检测模型。9.根据权利要求8所述的系统，其特征在于，所述融合子单元包括：连接模块，用于将所述校正后的自然光成像图片和校正后的红外线成像图片分别分成第一预设数量块，并一一对应按通道连接，得到第一预设数量个通道叠加图片块；自注意力机制操作模块，用于将所述第一预设数量个通道叠加图片块进行自注意力机制操作，得到第一预设数量个特征图中每个特征值的相关性；其中，所述相关性为特征图中一个位置的特征值与其他位置的特征值的关系；卷积操作模块，用于根据所述第一预设数量个特征图中每个特征值的相关性进行卷积操作，以使所述第一预设数量个通道叠加图片块的通道与所述校正后的自然光成像图片或校正后的红外线成像图片的通道相等，得到第一预设数量个融合特征块。10.根据权利要求8所述的系统，其特征在于，所述检测子单元包括：拼接模块，用于将所述第一预设数量个融合特征块进行拼接，得到多源特征图；第一分块模块，用于将所述多源特征图划分为第二预设数量块，并将每一块进行自注意力机制操作，得到第二预设数量个自注意力特征图；将所述第二预设数量个自注意力特征图进行融合并进行全卷积操作得到第一块自注意力特征图；第二分块模块，用于将所述第一块自注意力特征图划分为第三预设数量块，并将每一块进行自注意力机制操作，得到第三预设数量个自注意力特征图；将所述第三预设数量个自注意力特征图进行融合并进行全卷积操作得到第二块自注意力特征图；第三分块模块，用于将所述第二块自注意力特征图划分为第四预设数量块，并将每一块进行自注意力机制操作，得到第四预设数量个自注意力特征图；将所述第四预设数量个自注意力特征图进行融合并进行全卷积操作得到第三块自注意力特征图；融合模块，用于将所述第一块自注意力特征图、所述第二块自注意力特征图、所述第三块自注意力特征图依次进行卷积操作、融合操作得到总自注意力特征图；将所述总自注意力特征图输入到所述模型的检测头中进行检测，得到特征图中磁芯缺陷的位置以及所述当前轮磁芯缺陷检测模型；根据所述特征图中磁芯缺陷的位置与所述校正图片中磁芯缺陷的位置计算损失值。
技术总结
本发明公开一种基于多源特征的磁芯缺陷检测模型训练方法及系统，其中，该方法包括：S101，获取目标磁芯的自然光成像图片和红外线成像图片并作为训练集进行模型训练，得到当前轮磁芯缺陷检测模型；所述模型训练包括：将两个图片进行位置校正，得到校正后的两个图片；将校正后的两个图片分别分成第一预设数量块，并一一对应进行融合，得到第一预设数量个融合特征块；可得到不同成像之间的关系，解决单一成像的局限性；将第一预设数量个融合特征块进行拼接、分块操作，降低缺陷误检和漏检，采用自注意力机制操作和全卷积操作，得到的特征图具有全局性。并且计算得到当前轮的损失值；S102，重复S101直至损失值在预设范围内波动，得到目标磁芯缺陷检测模型。标磁芯缺陷检测模型。标磁芯缺陷检测模型。
技术研发人员：吴显德 李军 邹凡 鲁杰 雷明根
受保护的技术使用者：浙江华是科技股份有限公司
技术研发日：2022.07.21
技术公布日：2022/9/20