4 个月前

基于图注意力网络的超像素图像分类

基于图注意力网络的超像素图像分类

摘要

本文提出了一种基于图神经网络(GNN)模型的图像分类方法。我们将输入图像转换为区域邻接图(RAGs),其中区域由超像素组成,边连接相邻的超像素。实验结果表明,结合了图卷积和自注意力机制的图注意力网络(GATs)优于其他GNN模型。尽管由于在生成RAG过程中存在信息损失,原始图像分类器的表现优于GATs,但我们的方法为超越矩形网格图像的深度学习研究开辟了一个有趣的方向,例如360度全景视图。当前最先进的方法中的传统卷积核无法处理全景图,而适应后的超像素算法及其生成的区域邻接图可以自然地输入到GNN中,且不会出现拓扑问题。

基准测试

基准方法指标
superpixel-image-classification-on-75GAT
Classification Error: 3.81

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于图注意力网络的超像素图像分类 | 论文 | HyperAI超神经