基于空间注意力与图卷积的多标签图像分类算法 |
| |
引用本文: | 康萍萍,侯进,周浩然,陈子锐,李晨.基于空间注意力与图卷积的多标签图像分类算法[J].微电子学与计算机,2022(5):10-19. |
| |
作者姓名: | 康萍萍 侯进 周浩然 陈子锐 李晨 |
| |
作者单位: | 1. 西南交通大学计算机与人工智能学院;2. 西南交通大学综合交通大数据应用技术国家工程实验室;3. 西南交通大学信息科学与技术学院智能感知智慧运维实验室 |
| |
基金项目: | 四川省科技计划项目(2020SYSY0016);;国家重点研发计划(2020YFB1711902); |
| |
摘 要: | 针对传统多标签图像分类模型存在难以生成更接近相关标签的高层图像特征,以及因未能利用标签之间的视觉相关性而导致的识别精度不够高等问题,提出了一种基于空间注意力与图卷积的多标签图像分类算法.首先,利用图卷积网络学习标签邻接图特征和使用GLOVE算法,从标签序列获取的标签嵌入;其次,在高层语义信息中引入改进的空间注意力网络以...
|
关 键 词: | 图卷积网络 多标签图像分类 空间注意力 特征融合 |
|
|