010-61503882

首页 > Model Zoo > mixer_b16_224-Pytorch
mixer_b16_224-Pytorch
图像 视觉 分类
2024-06-04 | 2992人看过
模型信息

Mlp-Mixer,一个简单的2021年初,Google AI 团队续ViT模型之后回归到传统的MLP网络,基于MLP网络设计了一个全MLP的Mixer结构来进行计算机视觉任务。相较于过去被广泛适用于各个领域的基于卷积的CNN和最近大火的基于自注意力的Transformer而言,Mixer完全基于早期的多层感知器MLP,仅仅依赖于基本的矩阵乘法来完成数据的处理和特征的提取,通过单纯的MLP网络、非线性激活函数、Layer归一化也能够取得与CNN和ViT网络相似的性能。

论文:[《Mlp-Mixer》](https://arxiv.org/pdf/2105.01601.pdf)

开源模型链接:https://github.com/huggingface/pytorch-image-models/blob/v0.6.12/timm/models/mlp_mixer.py

数据集(ImageNet):http://www.image-net.org/


发表评论
全部评论(0)