加载中...
首页专利查询专利详情

*来源于国家知识产权局数据,仅供参考,实际以国家知识产权局展示为准

一种基于自注意力机制的图像分类方法

发明专利有效专利
  • 申请号:
    CN202110723547.4
  • IPC分类号:G06K9/62;G06N3/04;G06N3/08
  • 申请日期:
    2021-06-29
  • 申请人:
    沈阳雅译网络技术有限公司
著录项信息
专利名称一种基于自注意力机制的图像分类方法
申请号CN202110723547.4申请日期2021-06-29
法律状态实质审查申报国家中国
公开/公告日2021-09-10公开/公告号CN113378973A
优先权暂无优先权号暂无
主分类号G06K9/62IPC分类号G;0;6;K;9;/;6;2;;;G;0;6;N;3;/;0;4;;;G;0;6;N;3;/;0;8查看分类表>
申请人沈阳雅译网络技术有限公司申请人地址
辽宁省沈阳市和平区三好街78号东软电脑城C座11层 变更 专利地址、主体等相关变化,请及时变更,防止失效
权利人沈阳雅译网络技术有限公司当前权利人沈阳雅译网络技术有限公司
发明人杨木润;赵闯
代理机构沈阳新科知识产权代理事务所(特殊普通合伙)代理人李晓光
摘要
本发明提供一种基于自注意力机制的图像分类方法,步骤为:构建包含自注意力机制的Transformer模型,针对图像分类任务添加分类器单元;处理公开数据集ImageNet,调整原图片至合适大小;将调整后的图片划分成固定大小的子图,连接各子图后进行维度调整,得到图片嵌入向量;进行二维位置编码,得到二维位置编码向量,和图片嵌入向量连接,作为模型输入;将连接后向量送入Transformer模型,提取图片特征,最终解码时通过分类器单元将模型输出的向量转换成概率表示,完成图像分类。本发明通过自注意力机制的使用,能够有效地从图片中提取全局信息,即传统卷积神经网络所提取的图片特征,基于提取的特征能够有效地完成对图片的分类。

我浏览过的专利

专利服务由北京酷爱智慧知识产权代理公司提供