[发明专利]训练模型的方法、装置、电子设备、存储介质及产品

权利要求书

1.一种训练模型的方法，包括：

获取第一图像集，并调用第一模型以及第二模型，通过所述第一模型和所述第二模型分别对所述第一图像集进行目标检测，所述第一模型的目标检测准确度小于所述第二模型的目标检测准确度；

确定所述第一模型对所述第一图像集进行目标检测的第一目标检测结果，并确定所述第二模型对所述第一图像集进行目标检测的第二目标检测结果；

基于所述第一目标检测结果和所述第二目标检测结果，确定误检图像；

生成包括所述误检图像的第二图像集，并基于所述第二图像集训练所述第一模型。

2.根据权利要求1所述的方法，其中，所述生成包括所述误检图像的第二图像集，包括：

在目标数据库中，获取多张目标图像，所述目标图像中包括有所述第一模型和所述第二模型进行目标检测的检测目标；

将所述多张目标图像与所述误检图像进行组合，生成包括组合后图像的第二图像集。

3.根据权利要求2所述的方法，其中，所述将所述多张目标图像与所述误检图像进行组合，包括：

对所述误检图像进行预处理，得到误检区域图像；

对所述多张目标图像进行预处理，得到多张目标区域图像；

组合所述误检区域图像以及所述多张目标区域图像，得到组合后图像，其中，所述组合后图像的中心为所述误检区域图像，所述多张目标区域图像位于所述误检区域图像周边。

4.根据权利要求3所述的方法，其中，所述误检区域图像与所述多张目标区域图像以特定方式呈现，所述特定方式包括九宫格形式、十字形式和X形式。

5.根据权利要求3所述的方法，其中，所述对所述误检图像进行预处理，得到误检区域图像，包括：

扩展所述误检图像的尺寸，并在所述误检图像中确定误检区域；

在扩展尺寸后的误检图像中裁剪得到包括误检区域的误检区域图像。

6.根据权利要求1-5中任意一项所述的方法，其中，所述基于所述第一目标检测结果和所述第二目标检测结果，确定误检图像，包括：

确定第一模型对第一图像集中第一图像的第一目标检测结果，并确定所述第二模型对所述第一图像的第二目标检测结果；

确定所述第一目标检测结果和第二目标检测结果的交并比；

若所述交并比小于阈值，则将所述第一图像确定为误检图像。

7.一种训练模型的装置，包括：

目标检测单元，用于获取第一图像集，并调用第一模型以及第二模型，通过所述第一模型和所述第二模型分别对所述第一图像集进行目标检测，所述第一模型的目标检测准确度小于所述第二模型的目标检测准确度；

确定单元，用于确定所述第一模型对所述第一图像集进行目标检测的第一目标检测结果，并确定所述第二模型对所述第一图像集进行目标检测的第二目标检测结果，还用于基于所述第一目标检测结果和所述第二目标检测结果，确定误检图像；

生成单元，用于生成包括所述误检图像的第二图像集；

训练单元，用于基于所述第二图像集训练所述第一模型。

8.根据权利要求7所述的装置，其中，所述生成单元，还用于：

在目标数据库中，获取多张目标图像，所述目标图像中包括有所述第一模型和所述第二模型进行目标检测的检测目标；

将所述多张目标图像与所述误检图像进行组合，生成包括组合后图像的第二图像集。

9.根据权利要求8所述的装置，其中，所述生成单元，还用于：

对所述误检图像进行预处理，得到误检区域图像；

对所述多张目标图像进行预处理，得到多张目标区域图像；

组合所述误检区域图像以及所述多张目标区域图像，得到组合后图像，其中，所述组合后图像的中心为所述误检区域图像，所述多张目标区域图像位于所述误检区域图像周边。