[发明专利]图像多标签分类的方法、装置、设备及存储介质

摘要

本申请提出一种图像多标签分类的方法、装置、设备及存储介质，该方法包括：获取训练集，训练集包括标注有多个分类标签的样本图像；构建用于多标签分类的网络模型结构，网络模型结构包括多个激活层，激活层的数目与分类标签的数目相等；根据训练集训练网络模型结构，得到训练好的多标签分类模型。本申请使用具有多个激活层的多标签分类模型进行多标签分类。多标签分类模型的结构简单，运算量小。进一步地，在多标签分类模型中添加多头注意力机制，从而能学习到不同分类标签之间的关联性，提高多标签分类的准确性。多标签分类模型具有更轻的网络结构，训练时能更快收敛，提高模型训练效率，在相同资源情况下该多标签分类模型具有更高的服务性能。