本申请涉及图像识别技术领域,具体而言,涉及一种网络模型训练及电感缺陷识别方法、装置、电子设备。
背景技术:
目前国内外对于电感缺陷的检测,大多数是依赖人工检测。由于电感的体积较小,缺陷不太明显,需要在放大镜下进行检测。缺乏客观性、效率低下且劳动强度大。且由于电感缺陷的复杂性和缺陷之间的差异性较小的原因,传统的机器视觉技术难以满足要求。
针对上述问题,目前尚未有有效的技术解决方案。
技术实现要素:
本申请实施例的目的在于提供一种网络模型训练及电感缺陷识别方法、装置、电子设备,可以提高电感缺陷识别效率以及准确度。
第一方面,本申请实施例提供了一种网络模型训练方法,用于电感缺陷识别,所述方法包括:
获取源域数据以及目标域数据,其中,所述目标域数据包括多张具有缺陷的电感图片,所述源域数据包括多张非电感图片;
获取预设网络结构g,并根据所述源域数据对所述预设网络结构g进行初始化训练,以得到具有模型参数的目标网络结构g1;
基于所述目标域数据对所述目标网络结构g1进行训练,以得到目标网络模型g2。
可选地,在本申请实施例所述的网络模型训练方法中,所述获取源域数据以及目标域数据包括:
获取初始源域数据以及初始目标域数据,其中,所述初始目标域数据包括多张具有缺陷的电感图片,所述初始源域数据包括多张非电感图片;
对所述初始源域数据以及所述初始目标域数据内的图片尺寸进行统一化处理,以得到源域数据以及目标域数据,所述源域数据以及所述目标域数据内的图片尺寸相同。
可选地,在本申请实施例所述的网络模型训练方法中,所述获取源域数据以及目标域数据包括:
获取初始源域数据以及初始目标域数据,其中,所述初始目标域数据包括多张具有缺陷的电感图片,所述初始源域数据包括多张非电感图片;
对所述初始源域数据以及所述初始目标域数据内的图片尺寸进行统一化处理,以得到源域数据以及目标域数据,所述源域数据以及所述目标域数据内的图片尺寸相同。
可选地,在本申请实施例所述的网络模型训练方法中,所述基于所述目标域数据对所述目标网络结构g1进行训练,以得到目标网络模型g2之前,还包括:
对所述目标域数据内的电感图片依次进行预处理,以得到经过预处理的目标域数据,其中,所述预处理包括白化处理、翻转处理以及裁剪处理中的至少一种处理方式。
可选地,在本申请实施例所述的网络模型训练方法中,所述基于所述目标域数据对所述目标网络结构g1进行训练,以得到目标网络模型g2之前,还包括:
对所述目标域数据内的电感图片依次进行预处理,以得到经过预处理的目标域数据,其中,所述预处理包括白化处理、翻转处理以及裁剪处理中的至少一种处理方式。
可选地,在本申请实施例所述的网络模型训练方法中,所述预设网络结构g为预设层的残差网络结构。
第二方面,本申请实施例还提供了一种电感缺陷识别方法,所述方法采用上述任一项所述的方法生成的目标网络模型g2,所述方法包括:
将待识别的电感的图片输入所述目标网络模型g2,以输出电感缺陷的概率属性值;
根据所述概率属性值对所述待识别的电感的缺陷属性进行识别。
可选地,在本申请实施例所述的电感缺陷识别方法中,所述目标网络模型g2包括n个输出通道,其中,每一所述输出通道用于输出一种类别的电感缺陷的概率属性值,n为正整数;
而所述根据所述概率属性值对所述待识别的电感的缺陷属性进行识别,包括:
将所述目标网络模型g2输出的n个概率属性值分别与电感缺陷概率属性阈值进行比较;
若所述n个概率属性值中存在大于所述电感缺陷概率属性阈值的概率属性值,则所述待识别的电感具有对应类型的缺陷属性。
第三方面,本申请实施例还提供了一种网络模型训练方法训练装置,所述装置包括:
第一获取模块,用于获取源域数据以及目标域数据,其中,所述目标域数据包括多张具有缺陷的电感图片,所述源域数据包括多张非电感图片;
初始化模块,用于获取预设网络结构g,并根据所述源域数据对所述预设网络结构g进行初始化训练,以得到具有模型参数的目标网络结构g1;
训练模块,用于基于所述目标域数据对所述目标网络结构g1进行训练,以得到目标网络模型g2。
第四方面,本申请实施例还提供了一种电感缺陷识别装置,所述装置采用上述任一项所述的方法生成的目标网络模型g2,所述装置包括:
输出模块,用于将待识别的电感的图片输入所述目标网络模型g2,以输出电感缺陷的概率属性值;
识别模块,用于根据所述概率属性值对所述待识别的电感的缺陷属性进行识别。
第五方面,本申请实施例提供一种电子设备,包括处理器以及存储器,所述存储器存储有计算机可读取指令,当所述计算机可读取指令由所述处理器执行时,运行如上述第一方面提供的所述方法中的步骤。
第六方面,本申请实施例提供一种存储介质,其上存储有计算机程序,所述计算机程序被处理器执行时运行如上述第一方面提供的所述方法中的步骤。
由上可知,本申请实施例通过获取源域数据以及目标域数据,所述目标域数据包括多张具有缺陷的电感图片,所述源域数据包括多张非电感图片;获取预设网络结构g,并根据所述源域数据对所述预设网络结构g进行初始化训练,以得到具有模型参数的目标网络结构g1;基于所述目标域数据对所述目标网络结构g1进行训练,以得到目标网络模型g2;从而实现对电感的缺陷属性的识别,可以实现电感缺陷的自动化识别,并且,由于采用了源域数据中的图片来对该预设网络结构g进行初始化训练,可以有效解决目标领域样本数据不足的缺点,可以提高检测的效率以及节约检测成本。
本申请的其他特征和优点将在随后的说明书阐述,并且,部分地从说明书中变得显而易见,或者通过实施本申请实施例了解。本申请的目的和其他优点可通过在所写的说明书、权利要求书、以及附图中所特别指出的结构来实现和获得。
附图说明
为了更清楚地说明本申请实施例的技术方案,下面将对本申请实施例中所需要使用的附图作简单地介绍,应当理解,以下附图仅示出了本申请的某些实施例,因此不应被看作是对范围的限定,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他相关的附图。
图1为本申请实施例提供的网络模型训练方法的一种流程图。
图2为本申请实施例提供的电感缺陷识别方法的一种流程图。
图3为本申请实施例提供的网络模型训练装置的一种结构图。
图4为本申请实施例提供的电感缺陷识别装置的一种结构图。
图5为本申请实施例提供的电子设备的一种结构图。
具体实施方式
下面将结合本申请实施例中附图,对本申请实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本申请一部分实施例,而不是全部的实施例。通常在此处附图中描述和示出的本申请实施例的组件可以以各种不同的配置来布置和设计。因此,以下对在附图中提供的本申请的实施例的详细描述并非旨在限制要求保护的本申请的范围,而是仅仅表示本申请的选定实施例。基于本申请的实施例,本领域技术人员在没有做出创造性劳动的前提下所获得的所有其他实施例,都属于本申请保护的范围。
应注意到:相似的标号和字母在下面的附图中表示类似项,因此,一旦某一项在一个附图中被定义,则在随后的附图中不需要对其进行进一步定义和解释。同时,在本申请的描述中,术语“第一”、“第二”等仅用于区分描述,而不能理解为指示或暗示相对重要性。
请参照图1,图1是本本申请一些实施例中的网络模型训练方法的流程图。该网络模型训练方法,该网络模型用于电感缺陷识别,所述方法包括:
s101、获取源域数据以及目标域数据,其中,所述目标域数据包括多张具有缺陷的电感图片,所述源域数据包括多张非电感图片。
s102、获取预设网络结构g,并根据所述源域数据对所述预设网络结构g进行初始化训练,以得到具有模型参数的目标网络结构g1。
s103、基于所述目标域数据对所述目标网络结构g1进行训练,以得到目标网络模型g2。
其中,在该步骤s101中,该源域数据可以取自imagenet数据集,其包括多张非电感图片。该目标域数据为预先准备的多张具有缺陷的电感图片。其中,该多张具有缺陷的电感图片对应的电感缺陷有至少一种,在本实施例中电感缺陷类别有四种,该四种缺陷分别为:翼片超出凹槽、底面有翼、顶部无翼和翼片缺损。其中,该目标域数据中的图片的分辨率大小可以为440*422,为24位图,为bmp格式的图片,当然,其并不限于这种规格,也可以采用其他格式的图片、其他位的图,分辨率也可以调整。
在一些实施例中,为了便于后续的网络的训练,需要将该源域数据以及目标域数据中的图片调整为尺寸相同的图片。
该步骤s101包括以下子步骤:s1011、获取初始源域数据以及初始目标域数据,其中,所述初始目标域数据包括多张具有缺陷的电感图片,所述初始源域数据包括多张非电感图片;s1012、对所述初始源域数据以及所述初始目标域数据内的图片尺寸进行统一化处理,以得到源域数据以及目标域数据,所述源域数据以及所述目标域数据内的图片尺寸相同。
其中,可以对该初始源域数据的多张图片进行缩放处理,使其与初始目标域数据中的图片的尺寸相同。也可以对初始目标域数据的多张图片进行缩放处理,使其与初始源域数据的多张图片的尺寸相同。当然,可以理解地,也可以预先设定一个尺寸标准,根据该尺寸标准来对该初始源域数据以及初始目标域数据中的图片进行缩放调整,使得初始源域数据以及初始目标域数据中的图片均调整为该预设的尺寸标准。
在该步骤s102中,该预设网络结构g可以采用具有预设层的残差网络。具体地,在本实施例中,采用具有32层的残差网络。其中,该残差网络具有与缺陷类型数量对应的输出通道的数量。在该步骤s102中,首先判断该预设网络结构g是否已经经过初始化训练得到了模型参数,如果未初始化训练,则对该预设网络结构g进行初始化,并利用源域数据中的图片对该初始化后的预设网络结构g进行训练,并保存训练得到的权重等模型参数从而得到目标网络结构g1。
在该步骤s103中,采用加载该目标域数据中的多个电感图片到该目标网络结构g1中进行进一步优化训练,以得到目标网络模型g2。其中,在训练该训练步数train_steps=8000,权重衰减系数weight_decay=0.0002,数据增维大小padding_size=2,激活函数为relu,池化方法为maxpooling,优化器为momentumoptimizer,动量系数momentum=0.9,损失函数为cross-entropy,评价指标为train-loss、accuracy和混淆矩阵。
其中,预设网络结构g、目标网络结构g1、目标网络模型g2均包括n个输出通道,其中,每一所述输出通道用于输出一种类别的电感缺陷的概率属性值,n为正整数。例如,总共有4个通道。
在一些实施例中,在采用目标域数据中的图片进行训练的过程中,可以采用每一张图片中的电感缺陷仅包括一类的图片进行单通道专门训练。也即是说,例如,可以采用100张电感图片进行训练,其中,该100张电感图片中,有25张电感图片是a类缺陷,有25张电感图片是b类缺陷,有25张电感图片是c类缺陷,有25张电感图片是d类型缺陷,然后,可以分别采用这些缺陷类型固定的电感图片对对应的输出通道进行训练。
当然,可以理解地,也可以采用具有多种缺陷类型的电感的电感图片来同时训练该多个通道。
其中,在训练得到该目标网络模型g2的过程中,保存训练过程中的loss曲线,该loss曲线可以用来判断该目标网络模型g2的精度。
在一些实施例中,在执行该步骤s103之前,需要对该目标域数据的图片进行预处理,以提高准确率以及训练效率。具体地,在执行该步骤s103之前,还包括:对所述目标域数据内的电感图片依次进行预处理,以得到经过预处理的目标域数据,其中,所述预处理包括白化处理、翻转处理以及裁剪处理中的至少一种处理方式。
其中,可以采用以下公式进行白化处理:
其中,可以采用以下公式进行翻转处理:
其中,可以采用以下公式进行裁剪处理:
其中,μ代表平均值,δ2代表方差,pij代表像素点。
请参照图2,图2是本申请一些实施例中的一种电感缺陷识别方法的流程图。所述方法采用上述任一实施例所述的方法生成的目标网络模型g2,所述方法包括以下步骤:
s201、将待识别的电感的图片输入所述目标网络模型g2,以输出电感缺陷的概率属性值。
s202、根据所述概率属性值对所述待识别的电感的缺陷属性进行识别。
其中,在该步骤s201中,由于该目标网络模型g2包括n个输出通道,因此,在每次输入一张待识别的电感的图片到该目标网络模型g2中,就会输出n个概率属性值,每一个概率属性值用于描述一个对应类型的缺陷出现的概率。可以理解地,在一些实施例中,可以采用将该目标域数据中的图片按照预设比例划分成两组,一组图片用来训练得到该目标网络模型g2,另一组用来对该目标网络模型g2的准确度进行测试。
在该步骤s202中,在执行该步骤s202之前,需要预先设定一个概率属性阈值,该概率属性阈值为经验值。通过将该每个通道输出的概率属性值与该概率属性阈值进行比较,就可以判断出对应电感是否存在该类缺陷。例如,在本实施例中,该概率属性阈值ω=0.75。
在一些实施例中,该步骤s202包括以下步骤:s2021、将所述目标网络模型g2输出的n个概率属性值分别与电感缺陷概率属性阈值进行比较;s2022、若所述n个概率属性值中存在大于所述电感缺陷概率属性阈值的概率属性值,则所述待识别的电感具有对应类型的缺陷属性。
由上可知,本申请实施例通过获取源域数据以及目标域数据,所述目标域数据包括多张具有缺陷的电感图片,所述源域数据包括多张非电感图片;获取预设网络结构g,并根据所述源域数据对所述预设网络结构g进行初始化训练,以得到具有模型参数的目标网络结构g1;基于所述目标域数据对所述目标网络结构g1进行训练,以得到目标网络模型g2;从而实现对电感的缺陷属性的识别,可以实现电感缺陷的自动化识别,并且,由于采用了源域数据中的图片来对该预设网络结构g进行初始化训练,可以有效解决目标领域样本数据不足的缺点,可以提高检测的效率以及节约检测成本。
请参照图3,图3是本申请实施例还提供一种网络模型训练装置的结构图,所述装置包括:第一获取模块301、初始化模块302、训练模块303。
其中,该第一获取模块301用于获取源域数据以及目标域数据,其中,所述目标域数据包括多张具有缺陷的电感图片,所述源域数据包括多张非电感图片。该源域数据取自该imagenet数据集,其包括多张非电感图片。该目标域数据为预先准备的多张具有缺陷的电感图片。其中,该多张具有缺陷的电感图片对应的电感缺陷有至少一种,在本实施例中电感缺陷类别有四种,该四种缺陷分别为:翼片超出凹槽、底面有翼、顶部无翼和翼片缺损。其中,该目标域数据中的图片的分辨率大小可以为440*422,为24位图,为bmp格式的图片,当然,其并不限于这种规格,也可以采用其他格式的图片、其他位的图,分辨率也可以调整。
在一些实施例中,为了便于后续的网络的训练,需要将该源域数据以及目标域数据中的图片调整为尺寸相同的图片。
该第一获取模块301包括:第一获取单元,用于获取初始源域数据以及初始目标域数据,其中,所述初始目标域数据包括多张具有缺陷的电感图片,所述初始源域数据包括多张非电感图片;缩放处理单元,用于对所述初始源域数据以及所述初始目标域数据内的图片尺寸进行统一化处理,以得到源域数据以及目标域数据,所述源域数据以及所述目标域数据内的图片尺寸相同。
其中,可以对该初始源域数据的多张图片进行缩放处理,使其与初始目标域数据中的图片的尺寸相同。也可以对初始目标域数据的多张图片进行缩放处理,使其与初始源域数据的多张图片的尺寸相同。当然,可以理解地,也可以预先设定一个尺寸标准,根据该尺寸标准来对该初始源域数据以及初始目标域数据中的图片进行缩放调整,使得初始源域数据以及初始目标域数据中的图片均调整为该预设的尺寸标准。
其中,该初始化模块302用于获取预设网络结构g,并根据所述源域数据对所述预设网络结构g进行初始化训练,以得到具有模型参数的目标网络结构g1。该预设网络结构g可以采用具有预设层的残差网络。具体地,在本实施例中,采用具有32层的残差网络。其中,该残差网络具有与缺陷类型数量对应的输出通道的数量。首先对该预设网络结构g进行初始化,并利用源域数据中的图片对该初始化后的预设网络结构g进行训练,并保存训练得到的权重等模型参数从而得到目标网络结构g1。
其中,该训练模块303用于基于所述目标域数据对所述目标网络结构g1进行训练,以得到目标网络模型g2。在一些实施例中,在采用目标域数据中的图片进行训练的过程中,可以采用每一张图片中的电感缺陷仅包括一类的图片进行单通道专门训练。也即是说,例如,可以采用100张电感图片进行训练,其中,该100张电感图片中,有25张电感图片是a类缺陷,有25张电感图片是b类缺陷,有25张电感图片是c类缺陷,有25张电感图片是d类型缺陷,然后,可以分别采用这些缺陷类型固定的电感图片对对应的输出通道进行训练。
当然,可以理解地,也可以采用具有多种缺陷类型的电感的电感图片来同时训练该多个通道。
其中,在训练得到该目标网络模型g2的过程中,保存训练过程中的loss曲线,该loss曲线可以用来判断该目标网络模型g2的精度。
请参照图4,图4是本申请一些实施例中的一种电感缺陷识别装置的一种结构图。该装置包括:输出模块401以及识别模块402。
其中,该输出模块401用于将待识别的电感的图片输入所述目标网络模型g2,以输出电感缺陷的概率属性值;由于该目标网络模型g2包括n个输出通道,因此,在每次输入一张待识别的电感的图片到该目标网络模型g2中,就会输出n个概率属性值,每一个概率属性值用于描述一个对应类型的缺陷出现的概率。可以理解地,在一些实施例中,可以采用将该目标域数据中的图片按照预设比例划分成两组,一组图片用来训练得到该目标网络模型g2,另一组用来对该目标网络模型g2的准确度进行测试。
其中,该识别模块205用于根据所述概率属性值对所述待识别的电感的缺陷属性进行识别。该识别模块205需要预先设定一个概率属性阈值,该概率属性阈值为经验值。通过将该每个通道输出的概率属性值与该概率属性阈值进行比较,就可以判断出对应电感是否存在该类缺陷。例如,在本实施例中,该概率属性阈值ω=0.75。
目标网络模型g2包括n个输出通道,其中,每一所述输出通道用于输出一种类别的电感缺陷的概率属性值,n为正整数;而识别模块205包括:比较单元,用于将所述目标网络模型g2输出的n个概率属性值分别与电感缺陷概率属性阈值进行比较;识别单元,用于若所述n个概率属性值中存在大于所述电感缺陷概率属性阈值的概率属性值,则所述待识别的电感具有对应类型的缺陷属性。
由上可知,本申请实施例通过获取源域数据以及目标域数据,所述目标域数据包括多张具有缺陷的电感图片,所述源域数据包括多张非电感图片;获取预设网络结构g,并根据所述源域数据对所述预设网络结构g进行初始化训练,以得到具有模型参数的目标网络结构g1;基于所述目标域数据对所述目标网络结构g1进行训练,以得到目标网络模型g2;从而实现对电感的缺陷属性的识别,可以实现电感缺陷的自动化识别,并且,由于采用了源域数据中的图片来对该预设网络结构g进行初始化训练,可以有效解决目标领域样本数据不足的缺点,可以提高检测的效率以及节约检测成本。
请参照图5,图5为本申请实施例提供的一种电子设备的结构示意图,本申请提供一种电子设备5,包括:处理器501和存储器502,处理器501和存储器502通过通信总线503和/或其他形式的连接机构(未标出)互连并相互通讯,存储器502存储有处理器501可执行的计算机程序,当计算设备运行时,处理器501执行该计算机程序,以执行时执行上述实施例的任一可选的实现方式中的方法。
本申请实施例提供一种存储介质,所述计算机程序被处理器执行时,执行上述实施例的任一可选的实现方式中的方法。其中,存储介质可以由任何类型的易失性或非易失性存储设备或者它们的组合实现,如静态随机存取存储器(staticrandomaccessmemory,简称sram),电可擦除可编程只读存储器(electricallyerasableprogrammableread-onlymemory,简称eeprom),可擦除可编程只读存储器(erasableprogrammablereadonlymemory,简称eprom),可编程只读存储器(programmablered-onlymemory,简称prom),只读存储器(read-onlymemory,简称rom),磁存储器,快闪存储器,磁盘或光盘。
在本申请所提供的实施例中,应该理解到,所揭露装置和方法,可以通过其它的方式实现。以上所描述的装置实施例仅仅是示意性的,例如,所述单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,又例如,多个单元或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些通信接口,装置或单元的间接耦合或通信连接,可以是电性,机械或其它的形式。
另外,作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
再者,在本申请各个实施例中的各功能模块可以集成在一起形成一个独立的部分,也可以是各个模块单独存在,也可以两个或两个以上模块集成形成一个独立的部分。
在本文中,诸如第一和第二等之类的关系术语仅仅用来将一个实体或者操作与另一个实体或操作区分开来,而不一定要求或者暗示这些实体或操作之间存在任何这种实际的关系或者顺序。
以上所述仅为本申请的实施例而已,并不用于限制本申请的保护范围,对于本领域的技术人员来说,本申请可以有各种更改和变化。凡在本申请的精神和原则之内,所作的任何修改、等同替换、改进等,均应包含在本申请的保护范围之内。