前言
大家好,我是阿光。
本专栏整理了《PyTorch深度学习项目实战100例》,内包含了各种不同的深度学习项目,包含项目原理以及源码,每一个项目实例都附带有完整的代码+数据集。
正在更新中~ ✨
🚨 我的项目环境:
- 平台:Windows10
- 语言环境:python3.7
- 编译器:PyCharm
- PyTorch版本:1.8.1
💥 项目专栏:【PyTorch深度学习项目实战100例】
一、基于Pyramid Vision Transformer(PVT-v2)实现奥特曼识别
作为图像分类的第一个纯Transformer模型,ViT迈出了非常重要的一步,在ViT的基础上引入了金字塔结构是的它可以像ResNet模型那样无缝接入到各种下游任务中,本文尝试使用PVT-v2模型进行奥特曼的分类任务。
二、数据集介绍
本数据集使用的是几种不同的奥特曼图片,共有4种,赛文、迪迦、杰克、泰罗奥特曼
本数据集就不提供了,这个网上到处可见,可以自己去网上爬取自己喜欢的。