[发明专利]一种提高花型图分类打标准确率的方法

说明书

本发明公开了一种提高花型图分类打标准确率的方法，包括图片预处理模组、分类训练模组与数据集清洗再分类训练模型，所述图片预处理模组包括把图片等分成一组带有坐标的序列元素，所述分类训练模组包括有多个self‑attention、Norm和Feed Forward前馈网络组成。该提高花型图分类打标准确率的方法通过VIT模型训练其分类模型，通过其训练模型再去自动筛选掉数据集中类别定义模糊的图片，再在原始模型上训练迭代最新模型，对花型分类效果优于其他神经网络CNN分类模型。

技术领域

本发明涉及图片分类技术领域，尤其涉及一种提高花型图分类打标准确率的方法。

背景技术

图片分类技术已经经过了多年的发展，从最先色彩匹配、灰度匹配、到如今神经网络特征提取分类，而神经网络也经过了不同模型的提升，主要有这几大网络Lenet、Alexnet、VGGNet、GooLeNet、ResNet，图片分类的准确度得到了大幅度的提升。在花型网站中，对于图片正确的分类打标是非常重要的一环，正确的标签分类能够大大提高智能推荐给用户花型的准确率，也能间接提高花型搜索的响应时间，最重要的是能够让用户通过分类标签找到自己心仪的图片。

目前主流的图片分类方法是通过神经网络CNN方法，在已知数据集上，每个CNN通过提取图片一定范围的特征训练迭代得到分类模型，图片分类主流方法，在图片主体明确且训练资源较为丰富的情况下，图片分类结果较好。

但是现有的对于网站花型图，图片主题不明确，图片自带很多局部特征,花型图目标主体较多、特征视野分布在整张图上,传统神经网络分类方法对其分类效果并不佳。

发明内容

本发明的目的是为了解决现有技术中对于网站花型图，图片主题不明确，图片自带很多局部特征,花型图目标主体较多、特征视野分布在整张图上,传统神经网络分类方法对其分类效果并不佳的问题，而提出的一种提高花型图分类打标准确率的方法。

为了实现上述目的，本发明采用了如下技术方案：

一种提高花型图分类打标准确率的方法，包括图片预处理模组、分类训练模组与数据集清洗再分类训练模型，所述图片预处理模组包括把图片等分成一组带有坐标的序列元素，所述分类训练模组包括有多个self-attention、Norm和Feed Forward前馈网络组成。

优选地，所述图片预处理模组匹配ViT模型的输入，所述图片预处理模组包括H×W×C的三维图片转化为N×D的二维输入。

优选地，所述分类训练模组包括Transformer Encode。

优选地，所述数据集清洗再分类训练模型包括鉴别数据集中图片模块与筛选图片类别模块。

一种提高花型图分类打标准确率的方法，该方法的具体步骤如下：

步骤1-图片预处理：通过图片预处理模组把图片等分成一组带有坐标的序列元素，匹配vit模型的输入，首先把图片等比例划分成N片区域，每片区域长宽为P，输入序列为(N，P² c),再通过一个全连接层压缩为(N,D)的二维向量，图片割裂后，不再保有原来的位置信息，在Patch Embedding向量上增加一个Positional encoding Epos，在图片序列前增加一个和图片单个序列同纬度的可学习嵌入向量；

步骤2-分类训练模组：将图片序列经过多个self-attention再进行AddNorm，Add防止网络退化，Norm对每一层的激活值进行归一化，然后经过前馈网络，整个Encode经过N次该网络；

步骤3-数据集清洗再分类：在上一步生成的模型中，鉴别数据集中图片的具体类别及概率，再和图片所在类别进行对比，准确筛选图片类别，清洗原有数据集，在清洗的数据集上继续训练最新模型。

有益效果：