[发明专利]一种引入物体关系的图像分类方法

摘要

本发明公开了一种引入物体关系的图像分类方法，包括：步骤S1，通过注意力机制构建输入图像上的各个物体间的关系；步骤S2，基于所构建的各物体间的关系在输入图像上对各物体进行分类标注和位置标注；步骤S3，对所标注的分类特征图和位置特征图进行采样，并将采样得到的特征图作为注意力模型训练样本，训练得到注意力模型；步骤S4，通过注意力模型对输入图像进行目标分类识别。本发明利用注意力机制，构建场景图像中各物体间的物体关系，然后基于所构建的物体关系对物体特征进行图像标注，最后提取所标注的特征图像作为模型训练样本，训练得到注意力模型，并通过注意力模型对场景图像进行图像分类识别，提高了对目标分类识别检测的准确率。