本申请涉及缺陷检测技术领域，尤其涉及一种模型的训练方法、缺陷检测方法、装置、电子设备及介质，用以提高模型的训练效率。其中，方法包括：基于第一训练数据集，对初始检测模型中的分类网络的第二权重集进行循环迭代训练，获得多个第二缺陷类别的中间权重参数；基于各中间权重参数的平均值，对各中间权重参数进行更新，获得多个第二缺陷类别的目标权重参数；基于各目标权重参数对分类网络的第一权重集进行参数更新，更新后的第一权重集用于缺陷检测。由于本申请只需要通过模型训练对第二权重集进行更新后，基于第二权重集更新第一权重集，获得的第一权重集用于缺陷检测时，可以与使用全量的历史数据进行训练实现相当的效果，提高模型训练效率。
【技术实现步骤摘要】
所属的技术人员能够理解，本申请的各个方面可以实现为系统、方法或程序产品。因此，本申请的各个方面可以具体实现为以下形式，即：完全的硬件实施方式、完全的软件实施方式(包括固件、微代码等)，或硬件和软件方面结合的实施方式，这里可以统称为“电路”、“模块”或“系统”。与上述方法实施例基于同一专利技术构思，本申请实施例中还提供了一种电子设备。在一种实施例中，该电子设备可以是服务器，如图2所示的服务器。在该实施例中，电子设备的结构可以如图14所示，包括存储器1401，通讯模块1403以及一个或多个处理器1402。存储器1401，用于存储处理器1402执行的计算机程序。存储器1401可主要包括存储程序区和存储数据区，其中，存储程序区可存储操作系统，以及运行即时通讯功能所需的程序等；存储数据区可存储各种即时通讯信息和操作指令集等。存储器1401可以是易失性存储器(volatile memory)，例如随机存取存储器(random-access memory，ram)；存储器1401也可以是非易失性存储器(non-volatilememory)，例如只读存储器，快闪存储器(flash memory)，硬盘(hard disk drive，hdd)或固态硬盘(solid-state drive，ssd)；或者存储器1401是能够用于携带或存储具有指令或数据结构形式的期望的计算机程序并能够由计算机存取的任何其他介质，但不限于此。存储器1401可以是上述存储器的组合。处理器1402，可以包括一个或多个中央处理单元(central processing unit，cpu)或者为数字处理单元等等。处理器1402，用于调用存储器1401中存储的计算机程序时实现上述缺陷检测模型的训练方法。通讯模块1403用于与终端设备和其他服务器进行通信。本申请实施例中不限定上述存储器1401、通讯模块1403和处理器1402之间的具体连接介质。本申请实施例在图14中以存储器1401和处理器1402之间通过总线1404连接，总线1404在图14中以粗线描述，其它部件之间的连接方式，仅是进行示意性说明，并不引以为限。总线1404可以分为地址总线、数据总线、控制总线等。为便于描述，图14中仅用一条粗线描述，但并不描述仅有一根总线或一种类型的总线。存储器1401中存储有计算机存储介质，计算机存储介质中存储有计算机可执行指令，计算机可执行指令用于实现本申请实施例的缺陷检测模型的训练及应用方法。处理器1402用于执行上述的缺陷检测模型的训练方法，如图3所示。在另一种实施例中，电子设备也可以是其他电子设备，如图2所示的终端设备。在该实施例中，电子设备的结构可以如图15所示，包括：通信组件1510、存储器1520、显示单元1530、摄像头1540、传感器1550、音频电路1560、蓝牙模块1570、处理器1580等部件。通信组件1510用于与服务器进行通信。在一些实施例中，可以包括电路无线保真(wireless fidelity，wifi)模块，wifi模块属于短距离无线传输技术，电子设备通过wifi模块可以帮助用户收发信息。存储器1520可用于存储软件程序及数据。处理器1580通过运行存储在存储器1520的软件程序或数据，从而执行终端设备的各种功能以及数据处理。存储器1520可以包括高速随机存取存储器，还可以包括非易失性存储器，例如至少一个磁盘存储器件、闪存器件、或其他易失性固态存储器件。存储器1520存储有使得终端设备能运行的操作系统。本申请中存储器1520可以存储操作系统及各种应用程序，还可以存储执行本申请实施例缺陷检测模型的训练方法的计算机程序。显示单元1530还可用于显示由用户输入的信息或提供给用户的信息以及终端设备的各种菜单的图形用户界面(graphical user interface，gui)。具体地，显示单元1530可以包括设置在终端设备正面的显示屏1532。其中，显示屏1532可以采用液晶显示器、发光二极管等形式来配置。显示单元1530可以用于显示本申请实施例中的缺陷检测用户界面等。显示单元1530还可用于接收输入的数字或字符信息，产生与终端设备的用户设置以及功能控制有关的信号输入，具体地，显示单元1530可以包括设置在终端设备正面的触摸屏1531，可收集用户在其上或附近的触摸操作，例如点击按钮，拖动滚动框等。其中，触摸屏1531可以覆盖在显示屏1532之上，也可以将触摸屏1531与显示屏1532集成而实现终端设备的输入和输出功能，集成后可以简称触摸显示屏。本申请中显示单元1530可以显示应用程序以及对应的操作步骤。摄像头1540可用于捕获静态图像，用户可以将摄像头1540拍摄的图像通过应用发布评论。摄像头1540可以是一个，也可以是多个。物体通过镜头生成光学图像投射到感光元件。感光元件可以是电荷耦合器件(charge coupled device，ccd)或互补金属氧化物半导体(complementary metal-oxide-semiconductor，cmos)光电晶体管。感光元件把光信号转换成电信号，之后将电信号传递给处理器1580转换成数字图像信号。终端设备还可以包括至少一种传感器1550，比如加速度传感器1551、距离传感器1552、指纹传感器1553、温度传感器1554。终端设备还可配置有陀螺仪、气压计、湿度计、温度计、红外线传感器、光传感器、运动传感器等其他传感器。音频电路1560、扬声器1561、传声器1562可提供用户与终端设备之间的音频接口。音频电路1560可将接收到的音频数据转换后的电信号，传输到扬声器1561，由扬声器1561转换为声音信号输出。终端设备还可配置音量按钮，用于调节声音信号的音量。另一方面，传声器1562将收集的声音信号转换为电信号，由音频电路1560接收后转换为音频数据，再将音频数据输出至通信组件1510以发送给比如另一终端设备，或者将音频数据输出至存储器1520以便进一步处理。蓝牙模块1570用于通过蓝牙协议来与其他具有蓝牙模块的蓝牙设备进行信息交互。例如，终端设备可以通过蓝牙模块1570与同样具备蓝牙模块的可穿戴电子设备(例如智能手表)建立蓝牙连接，从而进行数据交互。处理器1580是终端设备的控制中心，利用各种接口和线路连接整个终端的各个部分，通过运行或执行存储在存储器1520内的软件程序，以及调用存储在存储器1520内的数据，执行终端设备的各种功能和处理数据。在一些实施例中，处理器1580可包括一个或多个处理单元；处理器1580还可以集成应用处理器和基带处理器，其中，应用处理器主要处理操作系统、用户界面和应用程序等，基带处理器主要处理无线通信。可以理解的是，上述基带处理器也可以不集成到处理器1580中。本申请中处理器1580可以运行操作系统、应用程序、用户界面显示及触控响应，以及本申请实施例的缺陷检测模型的训练及应用方法。另外，处理器1580与显示单元1530耦接。在一些可能的实施方式中，本申请提供的缺陷检测模型的训练及应用方法的各个方面还可以实现为一种程序产品的形式，其包括计算机程序，当程序产品在电子设备上运行时，计算机程序用于使电子设备执行本说明书上述描述的根据本申请各种本文档来自技高网...
【技术保护点】
1.一种模型的训练方法，其特征在于，所述方法包括：
2.如权利要求1所述的方法，其特征在于，所述第一训练数据集还包含从各缺陷图像集合中抽取的各历史缺陷图像；
3.如权利要求2所述的方法，其特征在于，所述基于所述平均值，以及所述各历史缺陷类别各自对应的历史缺陷图像的数量，分别对所述各历史缺陷类别各自的中间权重参数进行更新，获得所述各历史缺陷类别各自的目标权重参数，包括：
4.如权利要求3所述的方法，其特征在于，所述基于所述候选权重参数、所述一个历史缺陷类别对应的历史缺陷图像的数量，以及所述一个历史缺陷类别的历史权重参数，获得所述一个历史缺陷类别的目标权重参数，包括：
5.如权利要求4所述的方法，其特征在于，所述基于所述历史权重参数与样本重叠系数的乘积，与所述候选权重参数的和，获得所述一个历史缺陷类别的目标权重参数，包括：
6.如权利要求1所述的方法，其特征在于，所述第一训练数据集还包含从各缺陷图像集合中抽取的各历史缺陷图像；则通过以下方式获得所述各历史缺陷图像：
7.如权利要求6所述的方法，其特征在于，所述基于所述至少一个历史缺陷类别，各自对应的样本缺陷图像的数量，对所述至少一个历史缺陷类别各自对应的候选缺陷图像，进行图像抽取，获得所述各历史缺陷图像，包括：
8.如权利要求1～6任一项所述的方法，其特征在于，所述第一训练数据集还包含从各缺陷图像集合中抽取的各历史缺陷图像；
9.如权利要求1～6任一项所述的方法，其特征在于，所述基于各中间权重参数的平均值，对所述各中间权重参数进行更新，获得所述多个第二缺陷类别各自的目标权重参数，包括：
10.一种缺陷检测方法，其特征在于，所述方法包括：
11.一种模型的训练装置，其特征在于，包括：
12.一种缺陷检测装置，其特征在于，包括：
13.一种电子设备，其特征在于，其包括处理器和存储器，其中，所述存储器存储有计算机程序，当所述计算机程序被所述处理器执行时，使得所述处理器执行权利要求1～10中任一所述方法的步骤。
14.一种计算机可读存储介质，其特征在于，其包括计算机程序，当所述计算机程序在电子设备上运行时，所述计算机程序用于使所述电子设备执行权利要求1～10中任一所述方法的步骤。
15.一种计算机程序产品，其特征在于，包括计算机程序，所述计算机程序存储在计算机可读存储介质中；当电子设备的处理器从所述计算机可读存储介质读取所述计算机程序时，所述处理器执行所述计算机程序，使得所述电子设备执行权利要求1～10中任一所述方法的步骤。
...
【技术特征摘要】
1.一种模型的训练方法，其特征在于，所述方法包括：
2.如权利要求1所述的方法，其特征在于，所述第一训练数据集还包含从各缺陷图像集合中抽取的各历史缺陷图像；
3.如权利要求2所述的方法，其特征在于，所述基于所述平均值，以及所述各历史缺陷类别各自对应的历史缺陷图像的数量，分别对所述各历史缺陷类别各自的中间权重参数进行更新，获得所述各历史缺陷类别各自的目标权重参数，包括：
4.如权利要求3所述的方法，其特征在于，所述基于所述候选权重参数、所述一个历史缺陷类别对应的历史缺陷图像的数量，以及所述一个历史缺陷类别的历史权重参数，获得所述一个历史缺陷类别的目标权重参数，包括：
5.如权利要求4所述的方法，其特征在于，所述基于所述历史权重参数与样本重叠系数的乘积，与所述候选权重参数的和，获得所述一个历史缺陷类别的目标权重参数，包括：
6.如权利要求1所述的方法，其特征在于，所述第一训练数据集还包含从各缺陷图像集合中抽取的各历史缺陷图像；则通过以下方式获得所述各历史缺陷图像：
7.如权利要求6所述的方法，其特征在于，所述基于所述至少一个历史缺陷类别，各自对应的样本缺陷图像的数量，对所述至少一个历史缺陷类别各自对应的候选缺陷图像，进行图像抽取，获得所述各历史缺陷图...
【专利技术属性】
技术研发人员：曾怡，詹佳伟，刘俊，汪铖杰，
申请(专利权)人：重庆腾讯信息技术有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。