timm documentation
双胞胎
快速入门
教程
模型页面
模型系列结果
卷积网络
视觉 Transformer
BEiTCaiTConViTCrossViTDaViTDeiTEVAGCViTHieraHieraDet(SAM 2)MViTv2NaFlex ViTNesTPiTPVTv2Segment Anything ViTSwin TransformerSwin Transformer V2Swin Transformer V2 (CR)TNT双胞胎视觉 Transformer(相对位置)视觉 Transformer (ViT)VOLOXCiT
混合卷积 / 注意力网络
高效与移动网络
MLP 及其他 token 混合器
多模态模型的视觉编码器
参考
You are viewing main version, which requires installation from source. If you'd like
regular pip install, checkout the latest stable version (v1.0.31).
双胞胎
类别: ViT · 代码: timm/models/twins.py · 架构: 6 · 预训练权重: 6(6 个具有 ImageNet-1k 结果)
Twins-PCPVT(带有条件位置编码的金字塔视觉 Transformer)和 Twins-SVT(空间可分离自注意力,局部分组注意力与全局子采样注意力交替进行)。
论文
用法
请参阅快速入门了解如何加载预训练权重、进行图像预处理和微调,以及查阅特征提取了解如何将模型用作骨干网络。
>>> import timm
>>> model = timm.create_model('twins_pcpvt_small.in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)模型 API 支持
✓ 支持,✓ (square) 仅支持方形尺寸,✗ 不支持,– 无需支持(模型本身已接受任意输入尺寸)。通过在每个功能上运行随机初始化的模型(Twins 使用 twins_svt_small,尺寸为 224)进行验证,同一类的其他架构预计表现相同。详情请参阅 Model API 页面。
架构
| 架构 | 参数(M) | GMAC | 输入大小 | 预训练权重 |
|---|---|---|---|---|
twins_pcpvt_base | 43.83 | 6.68 | 224 | 1 |
twins_pcpvt_large | 60.99 | 9.84 | 224 | 1 |
twins_pcpvt_small | 24.11 | 3.83 | 224 | 1 |
twins_svt_base | 56.07 | 8.59 | 224 | 1 |
twins_svt_large | 99.27 | 15.15 | 224 | 1 |
twins_svt_small | 24.06 | 2.94 | 224 | 1 |
GMACs 和参数量对应于每种架构的默认输入大小和分类器,数据来自基准测试结果。
预训练权重
Top-1 是 results-imagenet.csv 中在所列评估规模下的 ImageNet-1k 验证准确率,- 表示该权重没有 ImageNet-1k 结果(例如预训练权重或图文权重)。Classes 是预训练分类器的大小,对于没有分类器的权重则为 none。
| 权重 | 输入大小 | Top-1(%) | 评估大小 | 类 | 许可证 | Hub |
|---|---|---|---|---|---|---|
twins_pcpvt_base.in1k | 224 | 82.724 | 224 | 1000 | apache-2.0 | timm |
twins_pcpvt_large.in1k | 224 | 83.158 | 224 | 1000 | apache-2.0 | timm |
twins_pcpvt_small.in1k | 224 | 81.116 | 224 | 1000 | apache-2.0 | timm |
twins_svt_base.in1k | 224 | 83.122 | 224 | 1000 | apache-2.0 | timm |
twins_svt_large.in1k | 224 | 83.718 | 224 | 1000 | apache-2.0 | timm |
twins_svt_small.in1k | 224 | 81.694 | 224 | 1000 | apache-2.0 | timm |