timm documentation
MetaFormer(PoolFormer、ConvFormer、CAFormer)
快速入门
教程
模型页面
模型系列结果
卷积网络
视觉 Transformer
混合卷积 / 注意力网络
BYOANet(BoTNet、HaloNet、LambdaNet)CoaTCSATv2FocalNet混合视觉 TransformerLeViTMaxViT 和 CoAtNetMetaFormer(PoolFormer、ConvFormer、CAFormer)Next-ViTTinyViTVisformerViTamin
高效与移动网络
MLP 及其他 token 混合器
多模态模型的视觉编码器
参考
MetaFormer(PoolFormer、ConvFormer、CAFormer)
类别: 混合 · 代码: timm/models/metaformer.py · 架构: 18 · 预训练权重: 50(其中 42 个包含 ImageNet-1k 结果)
MetaFormer 模型共享通用的 Transformer 块结构,并改变 token 混合器:PoolFormer / PoolFormerV2 使用池化,ConvFormer 使用可分离卷积,CAFormer 在早期阶段使用卷积,在后期阶段使用注意力机制。
论文
用法
请参阅快速入门了解如何加载预训练权重、进行图像预处理和微调,以及查阅特征提取了解如何将模型用作骨干网络。
>>> import timm
>>> model = timm.create_model('caformer_s18.sail_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)模型 API 支持
✓ 支持,✓ (square) 仅支持方形尺寸,✗ 不支持,– 无需支持(模型本身已接受任意输入尺寸)。通过在每个功能上运行随机初始化的模型(MetaFormer 使用 poolformerv2_s12,尺寸为 224)进行验证,预期同一类的其他架构表现相同。有关详细信息,请参阅 Model API 页面。
架构
| 架构 | 参数(M) | GMAC | 输入大小 | 预训练权重 |
|---|---|---|---|---|
caformer_b36 | 98.75 | 23.22 | 224 | 5 |
caformer_m36 | 56.20 | 13.29 | 224 | 5 |
caformer_s18 | 26.34 | 4.13 | 224 | 5 |
caformer_s36 | 39.30 | 8.00 | 224 | 5 |
convformer_b36 | 99.88 | 22.69 | 224 | 5 |
convformer_m36 | 57.05 | 12.89 | 224 | 5 |
convformer_s18 | 26.77 | 3.96 | 224 | 5 |
convformer_s36 | 40.01 | 7.67 | 224 | 5 |
poolformer_m36 | 56.17 | 8.80 | 224 | 1 |
poolformer_m48 | 73.47 | 11.59 | 224 | 1 |
poolformer_s12 | 11.92 | 1.82 | 224 | 1 |
poolformer_s24 | 21.39 | 3.41 | 224 | 1 |
poolformer_s36 | 30.86 | 5.00 | 224 | 1 |
poolformerv2_m36 | 56.08 | 8.81 | 224 | 1 |
poolformerv2_m48 | 73.35 | 11.59 | 224 | 1 |
poolformerv2_s12 | 11.89 | 1.83 | 224 | 1 |
poolformerv2_s24 | 21.34 | 3.42 | 224 | 1 |
poolformerv2_s36 | 30.79 | 5.01 | 224 | 1 |
GMACs 和参数量对应于每种架构的默认输入大小和分类器,数据来自基准测试结果。
预训练权重
Top-1 是 results-imagenet.csv 中在所列评估规模下的 ImageNet-1k 验证准确率,- 表示该权重没有 ImageNet-1k 结果(例如预训练权重或图文权重)。Classes 是预训练分类器的大小,对于没有分类器的权重则为 none。
| 权重 | 输入大小 | Top-1(%) | 评估大小 | 类 | 许可证 | Hub |
|---|---|---|---|---|---|---|
caformer_b36.sail_in1k | 224 | 85.502 | 224 | 1000 | apache-2.0 | timm |
caformer_b36.sail_in1k_384 | 384 | 86.400 | 384 | 1000 | apache-2.0 | timm |
caformer_b36.sail_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
caformer_b36.sail_in22k_ft_in1k | 224 | 87.450 | 224 | 1000 | apache-2.0 | timm |
caformer_b36.sail_in22k_ft_in1k_384 | 384 | 88.058 | 384 | 1000 | apache-2.0 | timm |
caformer_m36.sail_in1k | 224 | 85.242 | 224 | 1000 | apache-2.0 | timm |
caformer_m36.sail_in1k_384 | 384 | 86.164 | 384 | 1000 | apache-2.0 | timm |
caformer_m36.sail_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
caformer_m36.sail_in22k_ft_in1k | 224 | 86.606 | 224 | 1000 | apache-2.0 | timm |
caformer_m36.sail_in22k_ft_in1k_384 | 384 | 87.470 | 384 | 1000 | apache-2.0 | timm |
caformer_s18.sail_in1k | 224 | 83.664 | 224 | 1000 | apache-2.0 | timm |
caformer_s18.sail_in1k_384 | 384 | 85.036 | 384 | 1000 | apache-2.0 | timm |
caformer_s18.sail_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
caformer_s18.sail_in22k_ft_in1k | 224 | 84.092 | 224 | 1000 | apache-2.0 | timm |
caformer_s18.sail_in22k_ft_in1k_384 | 384 | 85.442 | 384 | 1000 | apache-2.0 | timm |
caformer_s36.sail_in1k | 224 | 84.528 | 224 | 1000 | apache-2.0 | timm |
caformer_s36.sail_in1k_384 | 384 | 85.762 | 384 | 1000 | apache-2.0 | timm |
caformer_s36.sail_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
caformer_s36.sail_in22k_ft_in1k | 224 | 85.780 | 224 | 1000 | apache-2.0 | timm |
caformer_s36.sail_in22k_ft_in1k_384 | 384 | 86.860 | 384 | 1000 | apache-2.0 | timm |
convformer_b36.sail_in1k | 224 | 84.826 | 224 | 1000 | apache-2.0 | timm |
convformer_b36.sail_in1k_384 | 384 | 85.770 | 384 | 1000 | apache-2.0 | timm |
convformer_b36.sail_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
convformer_b36.sail_in22k_ft_in1k | 224 | 86.992 | 224 | 1000 | apache-2.0 | timm |
convformer_b36.sail_in22k_ft_in1k_384 | 384 | 87.606 | 384 | 1000 | apache-2.0 | timm |
convformer_m36.sail_in1k | 224 | 84.486 | 224 | 1000 | apache-2.0 | timm |
convformer_m36.sail_in1k_384 | 384 | 85.580 | 384 | 1000 | apache-2.0 | timm |
convformer_m36.sail_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
convformer_m36.sail_in22k_ft_in1k | 224 | 86.152 | 224 | 1000 | apache-2.0 | timm |
convformer_m36.sail_in22k_ft_in1k_384 | 384 | 86.868 | 384 | 1000 | apache-2.0 | timm |
convformer_s18.sail_in1k | 224 | 82.968 | 224 | 1000 | apache-2.0 | timm |
convformer_s18.sail_in1k_384 | 384 | 84.392 | 384 | 1000 | apache-2.0 | timm |
convformer_s18.sail_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
convformer_s18.sail_in22k_ft_in1k | 224 | 83.768 | 224 | 1000 | apache-2.0 | timm |
convformer_s18.sail_in22k_ft_in1k_384 | 384 | 85.004 | 384 | 1000 | apache-2.0 | timm |
convformer_s36.sail_in1k | 224 | 84.068 | 224 | 1000 | apache-2.0 | timm |
convformer_s36.sail_in1k_384 | 384 | 85.380 | 384 | 1000 | apache-2.0 | timm |
convformer_s36.sail_in22k | 224 | - | - | 21841 | apache-2.0 | timm |
convformer_s36.sail_in22k_ft_in1k | 224 | 85.420 | 224 | 1000 | apache-2.0 | timm |
convformer_s36.sail_in22k_ft_in1k_384 | 384 | 86.374 | 384 | 1000 | apache-2.0 | timm |
poolformer_m36.sail_in1k | 224 | 82.104 | 224 | 1000 | apache-2.0 | timm |
poolformer_m48.sail_in1k | 224 | 82.462 | 224 | 1000 | apache-2.0 | timm |
poolformer_s12.sail_in1k | 224 | 77.238 | 224 | 1000 | apache-2.0 | timm |
poolformer_s24.sail_in1k | 224 | 80.294 | 224 | 1000 | apache-2.0 | timm |
poolformer_s36.sail_in1k | 224 | 81.428 | 224 | 1000 | apache-2.0 | timm |
poolformerv2_m36.sail_in1k | 224 | 82.214 | 224 | 1000 | apache-2.0 | timm |
poolformerv2_m48.sail_in1k | 224 | 82.612 | 224 | 1000 | apache-2.0 | timm |
poolformerv2_s12.sail_in1k | 224 | 78.008 | 224 | 1000 | apache-2.0 | timm |
poolformerv2_s24.sail_in1k | 224 | 80.746 | 224 | 1000 | apache-2.0 | timm |
poolformerv2_s36.sail_in1k | 224 | 81.576 | 224 | 1000 | apache-2.0 | timm |