[发明专利]分类模型构建方法及装置

摘要

本公开是关于一种分类模型构建方法及装置，其中，方法包括：获取至少一种物品类别对应的至少一个样本图片集；根据预设训练协议和预设网络模型对至少一个样本图片集中的样本图片进行训练，得到至少一种物品类别对应的分类模型，分类模型用于确定待分类图片所对应的目标物品类别。通过该技术方案，可以根据用户需要自动对用户相册中的照片进行分类，如将所有照片中包含猫、狗的照片分到宠物相册，从而无需用户手动进行图片分类，提升用户的使用体验，同时由于根据物品类别的图片集训练后得到的分类模型更符合物品类别的需求，可以提高图片识别的准确率，即提高识别的召回率，减小识别的错误率。

主权项

1.一种分类模型构建方法，其特征在于，包括：获取至少一种物品类别对应的至少一个样本图片集，其中，每个样本图片集对应一种物品类别，每个样本图片集中的每个样本图片中均包括该样本图片集所对应的物品类别下的物品；根据预设训练协议和预设网络模型对所述至少一个样本图片集中的样本图片进行训练，得到所述至少一种物品类别对应的分类模型，所述分类模型用于确定待分类图片所对应的目标物品类别；所述根据预设训练协议和预设网络模型对所述至少一个样本图片集中的样本图片进行训练，得到所述至少一种物品类别对应的分类模型，包括：保持所述预设网络模型的特征层的参数不变，对所述预设网络模型中的全连接层进行降维和优化处理，所述全连接层包括第一全连接层和第二全连接层，所述第一全连接层的输出为所述第二全连接层的输入；将所述第二全连接层的输出发送至所述预设网络模型的分类器层，对所述分类器层进行调整，得到所述至少一种物品类别对应的分类模型。