一种基于类激活映射机制的多标签图像分类方法和系统

摘要

本发明公开了一种基于类激活映射机制的多标签图像分类方法和系统，方法包括获取待分类图像，并将所述待分类图像转换为多维张量；将所述多维张量输入分类模型，得到待分类图像的分类结果；其中，本发明在模型训练阶段首先将每张图像的标签转化成标签词向量，然后学习不同标签之间的共现关系，并将这些关系融入到标签词向量中，因此能够解决现有图像分类方法没能充分学习标签之间的依赖关系，而导致图像分类效果不佳的技术问题；另外，由于本发明在模型训练阶段结合了类激活映射机制，从而维持了同一张图像的不同风格的视觉一致性，进而提升了模型的分类效果，可广泛应用于人工智能技术领域。