【PyTorch深度学习项目实战100例】—— 基于Pyramid Vision Transformer（PVT-v2）实现奥特曼识别 | 第58例

前言

大家好，我是阿光。

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

正在更新中~ ✨

我的项目环境：

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

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

---

一、基于Pyramid Vision Transformer（PVT-v2）实现奥特曼识别

作为图像分类的第一个纯Transformer模型，ViT迈出了非常重要的一步，在ViT的基础上引入了金字塔结构是的它可以像ResNet模型那样无缝接入到各种下游任务中，本文尝试使用PVT-v2模型进行奥特曼的分类任务。

二、数据集介绍

本数据集使用的是几种不同的奥特曼图片，共有4种，赛文、迪迦、杰克、泰罗奥特曼

本数据集就不提供了，这个网上到处可见，可以自己去网上爬取自己喜欢的。