- 博客(3)
- 收藏
- 关注
RSS订阅原创 基于transformer实现图像分类（pytorch实现）
对于初学者而言，手写transformer能够带来较为深刻的理解。今天我们来实现基于transformer的图像分类模型ViT。与传统transformer模型不同，在视觉任务中仅需完成判别式任务而非生成式任务，因此只需要实现编码器Encoder，不必实现解码器Decoder。除此之外还需要定义一个PatchEmbedding模块用于将图像转换为token。介绍完毕，下面开始进行实现。
2025-11-09 21:44:19
419
原创 pytorch实现卷积神经网络EfficientNetV1
实验开始前准备好pytorch，tensorboard，cuda等必须库，使用python train.py --data-path ' 你的文件路径开始'(其余参数由自己定义)
2025-09-26 10:22:42
216
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人