[发明专利]模型数据文件生成、图像识别方法、装置、设备及介质

说明书

本公开公开了一种模型数据文件生成、图像识别方法、装置、设备及介质。该方法包括：根据单张训练图像对初始的机器学习模型进行训练，所述单张训练图像中除目标物体之外为空白区域；将训练得到的与所述目标物体匹配的模型数据文件发送至客户端，以使客户端基于所述模型数据文件识别所述目标物体。本公开实施例可以加快模型生成速度，缩短模型训练周期。

技术领域

本公开实施例涉及数据处理技术，尤其涉及一种模型数据文件生成、图像识别方法、装置、设备及介质。

背景技术

随着机器学习技术的发展，机器学习模型已成为常用且准确的图像识别方法。

例如，在图像识别领域中，可以将采集的图像输入到预先训练的机器学习模型中，得到标注有目标物体的图像。若想模型能够识别出目标物体，需要收集包含该目标物体的多个图像，以及不包含该目标物体的多个图像，分别作为训练样本，甚至，还需要通过人工在各训练样本中进行标注。从而，对标注好的训练样本输入到模型中进行模型训练，将训练完成的模型作为图像识别模型投入使用。

上述模型生成的方法中，训练过程复杂，训练时间漫长。

发明内容

本公开实施例提供一种模型数据文件生成、图像识别方法、装置、设备及介质，可以加快模型生成速度，缩短模型训练周期。

第一方面，本公开实施例提供了一种模型数据文件生成方法，该方法包括：

根据单张训练图像对初始的机器学习模型进行训练，所述单张训练图像中除目标物体之外为空白区域；

将训练得到的与所述目标物体匹配的模型数据文件发送至客户端，以使客户端基于所述模型数据文件识别所述目标物体。

第二方面，本公开实施例还提供了一种图像识别方法，应用在客户端中，包括：

在接收到针对待检测图像的图像识别指令时，将至少一个模型数据文件加载到内存中；其中，所述模型数据文件通过根据单张训练图像对初始的机器学习模型进行训练形成；

基于各所述模型数据文件分别对所述待检测图像进行图像识别，得到所述待检测图像的图像识别结果。

第三方面，本公开实施例还提供了一种模型数据文件生成装置，该装置包括：

单样本训练模块，用于根据单张训练图像对初始的机器学习模型进行训练，所述单张训练图像中除目标物体之外为空白区域；

模型数据文件生成模块，用于将训练得到的与所述目标物体匹配的模型数据文件发送至客户端，以使客户端基于所述模型数据文件识别所述目标物体。

第四方面，本公开实施例还提供了一种图像识别装置，该装置包括：

模型数据文件加载模块，用于在接收到针对待检测图像的图像识别指令时，将至少一个模型数据文件加载到内存中；其中，所述模型数据文件通过根据单张训练图像对初始的机器学习模型进行训练形成；

图像识别模块，用于基于各所述模型数据文件分别对所述待检测图像进行图像识别，得到所述待检测图像的图像识别结果。

第五方面，本公开实施例还提供了一种电子设备，该电子设备包括：

一个或多个处理器；

存储器，用于存储一个或多个程序；

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如本公开实施例任一所述的模型数据文件生成方法或如本公开实施例任一所述的图像识别方法。

第六方面，本公开实施例还提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现如本公开实施例任一所述的模型数据文件生成方法或如本公开实施例任一所述的图像识别方法。