[发明专利]一种图像类别分类的训练方法及装置

说明书

本申请实施例公开了一种图像类别分类的训练方法及装置，用于提高数据分类的训练效率，节省系统资源。本申请实施例方法包括：获取预检测样本图像集；构建训练模型；将预检测样本图像集输入至训练模型确定正类及正类数量；通过训练模型将预检测样本图像集的图像数据平均分布于多张视觉处理器上；确定正类中心及负类中心；确定随机选取的负类中心的数量；根据随机选取的负类中心与正类中心计算总损失值；判断总损失值是否达到第一预设值；若否，则根据总损失值更新训练模型，对负类中心的数据进行二次随机选取，并根据二次选取的负类中心与正类中心二次计算总损失值，直至总损失值达到第一预设值时，确定训练模型训练完成。

技术领域

本申请实施例涉及智能安防领域，尤其涉及一种图像类别分类的训练方法及装置。

背景技术

随着当今社会的科学技术的发展，图像处理以及模式识别技术得到快速进步，基于视频图像处理的方式在实际生活中在很多领域得到实际应用。在实际应用中通常是通过收集大量数据完成训练，随着深度学习技术的发展和相关产品的落地，数据的种类和规模越来越大，以人脸识别为例，如今，绝大部分做人脸识别的团队或者公司都拥有几百万的账户或者千万级别的账户，上亿甚至是几十亿的数据，由于目前只能通过硬件的方式来对图像数据进行分类，而训练数据分类时应用到的视觉处理器(GPU，Graphics ProcessingUnit)数量以及显存又十分有限，使得训练这种千万级别账户的图像分类繁琐，训练效率低。

发明内容

本申请实施例提供了一种图像类别分类的训练方法及装置，用于提高数据分类的训练效率，节省系统资源。

本申请实施例在第一方面提供了一种图像类别分类的训练方法，包括：

获取预检测样本图像集；

构建训练模型，所述训练模型为设置有图像特征长度以及类别数量的神经网络模型；

将所述预检测样本图像集输入至所述训练模型确定正类及正类数量，所述正类为所述预检测样本图像集中的同类图像数据分布，所述正类的数量为所述预检测样本图像集中图像所属类别的类别数量；

通过所述训练模型将所述预检测样本图像集的图像数据平均分布于多张视觉处理器上；

根据所述正类在多张视觉处理器的平均分布的情况确定正类中心及负类中心；

根据所述正类的数量按照预设规则确定随机选取的负类中心的数量，所述负类中心为所述预检测样本图像集中任意两种不同类别的图像数据分布的中心；

根据所述随机选取的负类中心与所述正类中心计算总损失值；

判断所述总损失值是否达到第一预设值；

若否，则根据所述总损失值更新所述训练模型，对所述负类中心的数据进行二次随机选取，并根据二次选取的负类中心与所述正类中心二次计算总损失值，直至总损失值达到第一预设值时，则确定所述训练模型训练完成。

可选的，所述根据所述正类的数量按照预设规则确定随机选取的负类中心的数量，包括：

设置负类中心采样率；

确定视觉处理器上的图像数据的总类别个数，所述总类别个数为所述预检测样本图像集正类的个数；

根据所述负类中心采样率与所述总类别个数确定需要随机选取的负类中心的数量。

可选的，所述根据所述随机选取的负类中心与所述正类中心计算总损失值，包括：

计算所述随机选取的负类中心的负类损失值；

分别计算各个所述正类中心的正类损失值；

将所述负类损失值与各个正类损失值进行聚合，计算总损失值。