【PyTorch深度学习项目实战100例】—— 基于ViT（Vision_Transformer）识别七龙珠超级赛亚人 | 第52例

前言

大家好，我是阿光。

本专栏整理了《PyTorch深度学习项目实战100例》，内包含了各种不同的深度学习项目，包含项目原理以及源码，每一个项目实例都附带有完整的代码+数据集。

正在更新中~ ✨

我的项目环境：

• 平台：Windows10
• 语言环境：python3.7
• 编译器：PyCharm
• PyTorch版本：1.8.1

项目专栏：【PyTorch深度学习项目实战100例】

---

一、基于ViT（Vision_Transformer）识别七龙珠超级赛亚人

近几年Transformer打响NLP领域，内部采用了自注意力机制模块大大提高了高级语义特征的提取能力，随后有人提出使用Transformer进行图像分类，来利用注意力机制学习不同图像子块的相互影响。

本项目使用ViT网络模型进行七龙珠人物识别，ViT是一个采用注意力机制的识别网络，本项目迭代了10个epoch左右，准确率达到了95.99%，可见效果之棒。

二、数据集介绍

数据说明：