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

  • 8
  • 0
  • 约1.19万字
  • 约 10页
  • 2023-06-21 发布于四川
  • 举报

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

本发明提供一种基于自注意力机制的图像分类方法,步骤为:构建包含自注意力机制的Transformer模型,针对图像分类任务添加分类器单元;处理公开数据集ImageNet,调整原图片至合适大小;将调整后的图片划分成固定大小的子图,连接各子图后进行维度调整,得到图片嵌入向量;进行二维位置编码,得到二维位置编码向量,和图片嵌入向量连接,作为模型输入;将连接后向量送入Transformer模型,提取图片特征,最终解码时通过分类器单元将模型输出的向量转换成概率表示,完成图像分类。本发明通过自注意力机制的使用

(19)中华人民共和国国家知识产权局 (12)发明专利申请 (10)申请公布号 CN 113378973 A (43)申请公布日 2021.09.10 (21)申请号 202110723547.4 (22)申请日 2021.06.29 (71)申请人 沈阳雅译网络技术有限公司

文档评论(0)

1亿VIP精品文档

相关文档