基于GCN的多标签图像分类方法、模型构建方法及装置
公开
摘要
本发明公开了一种基于GCN的多标签图像分类方法、模型构建方法及装置,包括预处理模块、模型构建模块和训练模块;预处理模块,用于对已知数据集进行预处理,得到预处理后的数据集图像;训练模块,用于将预处理后的数据集图像输入多标签图像分类模型进行训练,以多个标签在图像上出现的概率为输出,得到训练好的多标签图像分类模型;模型构建模块,用于构建多标签图像分类模型并将其发送给训练模块。本发明结合了注意力机制模块和图卷积网络,通过对原始图像特征在空间和通道上获取注意力图以及通过图卷积网络获取更具鉴别意义的类别语义,有效加强了图像显著部分的特征和避免了感兴趣部分特征的丢失,提升了网络的输出精度。
基本信息
专利标题 :
基于GCN的多标签图像分类方法、模型构建方法及装置
专利标题(英):
暂无
公开(公告)号 :
CN114612681A
申请号 :
CN202210274122.4
公开(公告)日 :
2022-06-10
申请日 :
2022-03-20
授权号 :
暂无
授权日 :
暂无
发明人 :
赵国英任梅彭进业李展王琳赵万青杨文静
申请人 :
西北大学
申请人地址 :
陕西省西安市太白北路229号
代理机构 :
西安恒泰知识产权代理事务所
代理人 :
王芳
优先权 :
CN202210274122.4
主分类号 :
G06V10/44
IPC分类号 :
G06V10/44 G06V10/764 G06V10/82 G06K9/62 G06N3/04 G06N3/08
法律状态
2022-06-10 :
公开
注:本法律状态信息仅供参考,即时准确的法律状态信息须到国家知识产权局办理专利登记簿副本。
文件下载
暂无PDF文件可下载