[发明专利]模型处理方法及装置

说明书

本申请实施例涉及人工智能领域，公开一种模型处理方法及装置，该方法包括：从对象检测模型的训练样本集中获取目标训练样本，目标训练样本包括样本图像以及样本图像的N个标注对象框，N为正整数；确定在获取目标训练样本时，对象检测模型所处的当前训练阶段；基于训练阶段与对象框尺寸之间的适配策略，确定与当前训练阶段适配的目标对象框尺寸；根据每个标注对象框的对象框尺寸和目标对象框尺寸，从N个标注对象框中选取出M个样本对象框，样本对象框是指对象框尺寸大于或等于目标对象框尺寸的标注对象框，M为整数；采用样本图像以及M个样本对象框，对对象检测模型进行模型训练；可提高对象检测模型的检测准确率。

技术领域

本申请涉及人工智能领域，尤其涉及一种模型处理方法及装置。

背景技术

对象检测(Object Detection)，又称目标检测，其具体是指检测图像中是否包含目标对象(如人脸、手势等)的动作。目前，通常是采用对象检测模型来实现对图像的对象检测的；例如，针对人脸这一目标对象，通常是采用人脸检测模型对某图像进行人脸检测的。基于此，如何对对象检测模型进行训练优化，以提高对象检测模型对于目标对象的检测准确率是当前的热点研究方向。

发明内容

本申请实施例提供一种模型处理方法、装置、设备及存储介质，可提高对象检测模型的检测准确率。

一方面，本申请实施例提供了一种模型处理方法，包括：

从对象检测模型的训练样本集中获取目标训练样本，所述目标训练样本包括样本图像以及所述样本图像的N个标注对象框，N为正整数；

确定在获取所述目标训练样本时，所述对象检测模型所处的当前训练阶段；一个训练阶段包括：采用所述训练样本集中的全部数据对所述对象检测模型进行一次完整训练的过程；

基于训练阶段与对象框尺寸之间的适配策略，确定与所述当前训练阶段适配的目标对象框尺寸；

根据每个标注对象框的对象框尺寸和所述目标对象框尺寸，从所述N个标注对象框中选取出M个样本对象框，样本对象框是指对象框尺寸大于或等于所述目标对象框尺寸的标注对象框，M为整数；

采用所述样本图像以及所述M个样本对象框，对所述对象检测模型进行模型训练。

一方面，本申请实施例提供了一种模型处理装置，包括：

获取单元，用于从对象检测模型的训练样本集中获取目标训练样本，所述目标训练样本包括样本图像以及所述样本图像的N个标注对象框，N为正整数；

处理单元，用于确定在获取所述目标训练样本时，所述对象检测模型所处的当前训练阶段；一个训练阶段包括：采用所述训练样本集中的全部数据对所述对象检测模型进行一次完整训练的过程；

所述处理单元，还用于基于训练阶段与对象框尺寸之间的适配策略，确定与所述当前训练阶段适配的目标对象框尺寸；

所述处理单元，还用于根据每个标注对象框的对象框尺寸和所述目标对象框尺寸，从所述N个标注对象框中选取出M个样本对象框，样本对象框是指对象框尺寸大于或等于所述目标对象框尺寸的标注对象框，M为整数；

所述处理单元，还用于采用所述样本图像以及所述M个样本对象框，对所述对象检测模型进行模型训练。

一方面，本申请实施例提供了一种模型处理设备，其特征在于，所述模型处理设备包括输入接口和输出接口，还包括：

处理器，适于实现一条或多条指令；以及，

计算机存储介质，所述计算机存储介质存储有一条或多条指令，所述一条或多条指令适于由所述处理器加载并执行上述模型处理方法。