timm documentation
CSATv2
快速入门
教程
模型页面
模型系列结果
卷积网络
视觉 Transformer
混合卷积 / 注意力网络
BYOANet(BoTNet、HaloNet、LambdaNet)CoaTCSATv2FocalNet混合视觉 TransformerLeViTMaxViT 和 CoAtNetMetaFormer(PoolFormer、ConvFormer、CAFormer)Next-ViTTinyViTVisformerViTamin
高效与移动网络
MLP 及其他 token 混合器
多模态模型的视觉编码器
参考
CSATv2
类别: 混合 · 代码: timm/models/csatv2.py · 架构: 2 · 预训练权重: 3(0 个具有 ImageNet-1k 结果)
CSATv2 是一个频域视觉模型,它基于输入图像的 DCT 变换,并结合空间注意力机制。
论文
该模块的文档字符串未引用任何论文。
用法
请参阅快速入门了解如何加载预训练权重、进行图像预处理和微调,以及查阅特征提取了解如何将模型用作骨干网络。
>>> import timm
>>> model = timm.create_model('csatv2.r512_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)模型 API 支持
✓ 支持,✓ (square) 仅支持方形尺寸,✗ 不支持,– 无需支持(模型本身已接受任意输入尺寸)。通过在每个功能上运行随机初始化的模型(CSATv2 使用 csatv2,尺寸为 256)进行验证,同一类的其他架构预计表现相同。有关详细信息,请参阅 模型 API 页面。
架构
| 架构 | 参数(M) | GMAC | 输入大小 | 预训练权重 |
|---|---|---|---|---|
csatv2 | 11.10 | 1.39 | 512 | 1 |
csatv2_21m | 20.70 | 4.72 | 640 | 2 |
GMACs 和参数量对应于每种架构的默认输入大小和分类器,数据来自基准测试结果。
预训练权重
Top-1 是 results-imagenet.csv 中在所列评估规模下的 ImageNet-1k 验证准确率,- 表示该权重没有 ImageNet-1k 结果(例如预训练权重或图文权重)。Classes 是预训练分类器的大小,对于没有分类器的权重则为 none。
| 权重 | 输入大小 | Top-1(%) | 评估大小 | 类 | 许可证 | Hub |
|---|---|---|---|---|---|---|
csatv2.r512_in1k | 512 | - | - | 1000 | - | timm |
csatv2_21m.sw_r512_in1k | 512 | - | - | 1000 | - | timm |
csatv2_21m.sw_r640_in1k | 640 | - | - | 1000 | - | timm |