本申请公开了一种模型训练方法、使用方法、相关装置及存储介质，其中，标志特征提取模型训练方法包括：获取标志的第一样本数据集；将第一样本数据集输入到预设的网络模型中进行训练，生成第一标志特征提取模型；获取标志的第二样本数据集；根据第二样本数据集和第一标志特征提取模型进行小样本训练，得到第二标志特征提取模型。本申请在现有技术对不常见标志的分类识别准确率低，效果差基础上，通过包括不常见的目标标志类型的图像样本数据的两个不同样本数据集，依次进行模型训练，生成标志特征提取模型，在不增加数据集成本的基础上，充分利用小样本训练集进行训练，达到更高分类识别精度和快速响应的目的，同时兼顾了模型的可扩展性。型的可扩展性。型的可扩展性。
【技术实现步骤摘要】
模型训练方法、使用方法、相关装置及存储介质
[0001]本专利技术涉及通信
，具体涉及一种模型训练方法、使用方法、相关装置及存储介质。
技术介绍
[0002]深度学习作为图像识别算法中的主流方法，在大规模数据集的情况下拥有极好的分类效果。然而现实中需要使用到图像识别算法的项目一般难以支付大规模数据集的构建成本，甚至有时候数据总量本身就很少，例如某些标志类型本身就不常见，采集的样本数据非常少，无法收集到训练深度学习模型所需要的数据量，因此导致该标志类型分类识别准确率低，效果差。
技术实现思路
[0003]本专利技术实施例提供一种模型训练方法、使用方法、相关装置及存储介质，在不增加数据集成本的基础上，充分利用小样本训练集进行训练，达到更高分类识别精度和快速响应的目的，同时兼顾了模型的可扩展性。
[0004]一方面，本申请提供一种标志特征提取模型训练方法，所述标志特征提取模型训练方法包括：
[0005]获取标志的第一样本数据集，所述第一样本数据集中包括目标标志类型的图像样本数据，所述目标标志类型为小样本标志类型；
[0006]将所述第一样本数据集输入到预设的网络模型中进行训练，生成第一标志特征提取模型；
[0007]获取标志的第二样本数据集，所述第二样本数据集与所述第一样本数据集为同种类的样本数据集，所述第二样本数据集中的数据全部为所述目标标志类型的图像样本数据，所述第一样本数据集和所述第二样本数据集不同；
[0008]根据所述第二样本数据集和所述第一标志特征提取模型进行小样本训练，得到第二标志特征提取模型。
[0009]在本申请一些实施例中，所述第一样本数据集中包括多种标志类型的图像样本数据，所述多种标志类型中包括目标标志类型；所述第一标志特征提取模型中损失函数为交叉熵损失函数；
[0010]所述根据所述第二样本数据集和所述第一标志特征提取模型进行小样本训练，得到第二标志特征提取模型，包括：
[0011]将所述第一标志特征提取模型中损失函数调整为三元组损失函数，得到第三标志特征提取模型；
[0012]将所述第二样本数据集输入到所述第三标志特征提取模型中进行训练，生成第二标志特征提取模型。
[0013]在本申请一些实施例中，所述获取标志的第二样本数据集，包括：
[0014]从所述第一样本数据集中筛选目标标志类型的样本数据，得到所述样本数据集。
[0015]在本申请一些实施例中，所述获取标志的第二样本数据集，包括：
[0016]获取所述目标标志类型的小样本数据；
[0017]对所述小样本数据进行增广处理，得到第二样本数据集，所述第二样本数据集中存在至少300个所述目标标志类型图像。
[0018]在本申请一些实施例中，所述第一样本数据集中包括仅包括所述目标标志类型的图像样本数据；所述第一标志特征提取模型中损失函数为交叉熵损失函数；
[0019]所述根据所述第二样本数据集和所述第一标志特征提取模型进行小样本训练，得到第二标志特征提取模型，包括：
[0020]将所述第一标志特征提取模型中损失函数调整为三元组损失函数，得到第五标志特征提取模型；
[0021]将所述第二样本数据集输入到所述第五标志特征提取模型中进行训练，生成第二标志特征提取模型。
[0022]在本申请一些实施例中，所述预设的网络模型中包括所述检测标志区域的第一检测器；所述第一标志特征提取模型中包括检测标志区域的第二检测器；
[0023]所述根据所述第二样本数据集和所述第一标志特征提取模型进行小样本训练，得到第二标志特征提取模型，包括：
[0024]将预设的网络模型中的第一检测器，替换为所述第一标志特征提取模型中的第二检测器，得到第四标志特征提取模型；
[0025]将所述第二样本数据集输入到所述第四标志特征提取模型中进行训练，生成第二标志特征提取模型。
[0026]另一方面，本申请还提供一种模型使用方法，所述方法还包括：
[0027]获取待识别的标志图像；
[0028]将所述待识别的标志图像输入第二标志特征提取模型，以获取所述待识别的标志图像的第一分类特征。
[0029]在本申请一些实施例中，所述方法还包括：
[0030]建立多个特征库，所述多个特征库与多种标志类型对应；
[0031]在获取所述待识别的标志图像的第一分类特征之后，确定所述待识别的标志图像的标志类型；
[0032]若所述待识别的标志图像的标志类型为第一标志类型，则将所述第一分类特征保存在所述第一标志类型对应的第一特征库中。
[0033]在本申请一些实施例中，每种标志类型对应的特征库中包括已分类标志图像对应的至少一个第二分类特征；
[0034]所述确定所述待识别的标志图像的标志类型，包括：
[0035]分别计算所述第一分类特征与每种标志类型的特征库中保存的一个第二分类特征的相似值；
[0036]若所述第一相似值低于第一预设阈值，且所述第二相似值低于第二预设阈值，确定所述待识别的标志图像的标志类型为第一标志类型，所述第一相似值为所述待识别的标志图像的分类特征，与所述第一特征库中一个第二分类特征的相似值，所述第二相似值为
所述待识别的标志图像的分类特征，与目标特征库中一个第二分类特征的相似值。
[0037]另一方面，本申请提供一种标志识别方法，所述方法包括：
[0038]获取待识别的标志图像；
[0039]将所述待识别的标志图像输入第二标志特征提取模型，以获取所述待识别的标志图像的分类特征；
[0040]在获取所述待识别的标志图像的分类特征之后，确定所述待识别的标志图像的标志类型。
[0041]在本申请一些实施例中，多种标志类型对应多个特征库，每种标志类型对应的特征库中包括已分类标志图像对应的至少一个第二分类特征；
[0042]所述确定所述待识别的标志图像的标志类型，包括：
[0043]分别计算所述待识别的标志图像的分类特征，与每种标志类型的特征库中保存的一个第二分类特征的相似值；
[0044]若所述第一相似值低于第一预设阈值，且所述第二相似值低于第二预设阈值，确定所述待识别的标志图像的标志类型为第一标志类型，所述第一相似值为所述待识别的标志图像的分类特征，与所述第一特征库中一个第二分类特征的相似值，所述第二相似值为所述待识别的标志图像的分类特征，与目标特征库中一个第二分类特征的相似值。
[0045]另一方面，本申请提供一种标志特征提取模型训练装置，所述标志特征提取模型训练装置包括：
[0046]第一获取单元，用于获取标志的第一样本数据集，所述第一样本数据集中包括目标标志类型的图像样本数据，所述目标标志类型为小样本标志类型；
[0047]生成单元，用于将所述第一样本数据集输入到预设的网络模型中进行训练，生成第一标志特征提取模型；
[0048]第二获取单元，用于获取标志本文档来自技高网...
【技术保护点】
【技术特征摘要】
1.一种标志特征提取模型训练方法，其特征在于，所述标志特征提取模型训练方法包括：获取标志的第一样本数据集，所述第一样本数据集中包括目标标志类型的图像样本数据，所述目标标志类型为小样本标志类型；将所述第一样本数据集输入到预设的网络模型中进行训练，生成第一标志特征提取模型；获取标志的第二样本数据集，所述第二样本数据集与所述第一样本数据集为同种类的样本数据集，所述第二样本数据集中的数据全部为所述目标标志类型的图像样本数据，所述第一样本数据集和所述第二样本数据集不同；根据所述第二样本数据集和所述第一标志特征提取模型进行小样本训练，得到第二标志特征提取模型。2.根据权利要求1所述的标志特征提取模型训练方法，其特征在于，所述第一样本数据集中包括多种标志类型的图像样本数据，所述多种标志类型中包括目标标志类型；所述第一标志特征提取模型中损失函数为交叉熵损失函数；所述根据所述第二样本数据集和所述第一标志特征提取模型进行小样本训练，得到第二标志特征提取模型，包括：将所述第一标志特征提取模型中损失函数调整为三元组损失函数，得到第三标志特征提取模型；将所述第二样本数据集输入到所述第三标志特征提取模型中进行训练，生成第二标志特征提取模型。3.根据权利要求2所述的标志特征提取模型训练方法，其特征在于，所述获取标志的第二样本数据集，包括：从所述第一样本数据集中筛选目标标志类型的样本数据，得到所述样本数据集；或者，获取所述目标标志类型的小样本数据；对所述小样本数据进行增广处理，得到第二样本数据集，所述第二样本数据集中存在至少300个所述目标标志类型图像。4.根据权利要求1所述的标志特征提取模型训练方法，其特征在于，所述预设的网络模型中包括检测标志区域的第一检测器；所述第一标志特征提取模型中包括检测标志区域的第二检测器；所述根据所述第二样本数据集和所述第一标志特征提取模型进行小样本训练，得到第二标志特征提取模型，包括：将预设的网络模型中的第一检测器，替换为所述第一标志特征提取模型中的第二检测器，得到第四标志特征提取模型；将所述第二样本数据集输入到所述第四标志特征提取模型中进行训练，生成第二标志特征提取模型。5.一种模型使用方法，其特征在于，所述方法还包括：获取待识别的标志图像；将所述待识别的标志图像输入第二标志特征提取模型，以获取所述待识别的标志图像的第一分类特征。6.根据权利要求5所述的模型使用方法，其特征在于，所述方法还包括：
建立多...
【专利技术属性】
技术研发人员：冯昊楠，楚梦蝶，童星，
申请(专利权)人：顺丰科技有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。