[发明专利]一种对象识别模型的确定方法、装置、电子设备及存储介质

说明书

本申请实施例所公开的一种对象识别模型的确定方法、装置、电子设备及存储介质，其中，方法包括确定装置获取第一图片集合和第二图片集合，基于对象识别模型集合中的每个对象识别模型，确定第一图片集合中待识别对象集合的第二类别信息集合和第二位置数据集合，并从对象识别模型集合中确定对象识别模型，确定装置还基于对象识别模型，并从第二图片集合中确定目标图片集合，利用目标图片集合对对象识别模型进行训练，得到训练后的对象识别模型。基于本申请实施例，不仅可以从第二图片集合中筛选出标注质量较高的目标图片，提高用于训练对象识别模型的图片的干净程度，而且还可以改善对象识别模型的识别效果，提高毫米波图像中危险品的识别率。

技术领域

本发明涉及毫米波全息图像技术领域，尤其涉及一种对象识别模型的确定方法、装置、电子设备及存储介质。

背景技术

毫米波能够穿透衣物，对人体表面进行成像，且毫米波辐射属于非电离辐射，对人体无害，因此能够克服金属探测器无法检测陶瓷制品和X光射线容易对人体造成危害的缺陷，代替传统的金属探测器和X光射线，成为安检领域的主流技术手段，有效检测出藏匿在人体表层的可疑物品，如枪支、陶瓷刀具、不明液体、爆炸物等危险品。

在现有的毫米波全息图像识别技术中，主要依靠标注人员的眼睛对图像进行观察以判断被检测图像中是否存在目标危险品，然而由于毫米波图像的分辨率较低，标注人员很难保证标注的有效性，进而将会产生较多漏标或者错标的数据。针对上述问题，本领域技术人员训练一种模型以识别毫米波图像中的目标危险品，但当训练数据中存在较多脏数据时，模型将会误学习，把本不是目标危险品的数据认定为是目标危险品，如此，将直接导致模型的识别率降低。因此，有必要对模型的训练数据进行清洗，以减少错标或漏标的情况出现。

发明内容

本申请实施例提供一种对象识别模型的确定方法、装置、电子设备及存储介质，可以从第二图片集合中筛选出标注质量较高的目标图片，提高用于训练对象识别模型的图片的干净程度，改善对象识别模型的识别效果，提高毫米波图像中危险品的识别率。

本申请实施例提供的对象识别模型的确定方法，该方法包括：

获取第一图片集合和第二图片集合；第一图片集合和第二图片集合中含有待识别对象集合和待识别对象集合对应的第一类别信息集合和第一位置数据集合，

其中，待识别对象集合中的待识别对象与第一类别信息集合中的第一类别信息一一对应，待识别对象集合中的待识别对象与第一位置数据集合中的第一位置数据一一对应；

基于对象识别模型集合中的每个对象识别模型，确定第一图片集合中待识别对象集合的第二类别信息集合和第二位置数据集合；

根据第一类别信息集合、第二类别信息集合、第一位置数据集合和第二位置数据集合，从对象识别模型集合中确定对象识别模型；

基于对象识别模型，确定第二图片集合中待识别对象集合的第三类别信息集合和第三位置数据集合；

根据第一类别信息集合、第三类别信息集合、第一位置数据集合和第三位置数据集合，从第二图片集合中确定目标图片集合；

利用目标图片集合对对象识别模型进行训练，得到训练后的对象识别模型。

进一步地，根据第一类别信息集合、第二类别信息集合、第一位置数据集合和第二位置数据集合，从对象识别模型集合中确定对象识别模型，包括：

根据第一类别信息集合和第二类别信息集合确定第一分类差值集合；

根据第一位置数据集合和第二位置数据集合确定第一定位差值集合；

根据第一分类差值集合和第一定位差值集合确定第一差值集合；

从第一差值集合中确定数值最小的第一差值所对应的对象识别模型为对象识别模型。