timm documentation

XCiT

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

XCiT

类别: ViT · 代码: timm/models/xcit.py · 架构: 28 · 预训练权重: 42(42 个包含 ImageNet-1k 结果)

Cross-Covariance Image Transformers (XCiT) 用跨特征通道的交叉协方差注意力取代了 token 自注意力,其计算复杂度与 token 数量呈线性关系,并添加了局部 patch 交互层。

论文

用法

请参阅快速入门了解如何加载预训练权重、进行图像预处理和微调,以及查阅特征提取了解如何将模型用作骨干网络。

>>> import timm
>>> model = timm.create_model('xcit_small_12_p8_224.fb_dist_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

模型 API 支持

功能Xcit
features_only=True 特征图✓(步长 8, 8, 8)
forward_intermediates()✓
prune_intermediate_layers()✓
任意输入尺寸,无需更改✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
逐层学习率衰减(group_matcher())✓
reset_classifier()✓

✓ 支持,✓ (square) 仅支持方形尺寸,✗ 不支持,– 无需支持(模型本身已接受任意输入尺寸)。通过在每个功能上运行随机初始化的模型(Xcit 使用 xcit_nano_12_p8_224,尺寸为 224)进行验证,同一类的其他架构预计表现相同。有关详细信息,请参阅 模型 API 页面。

架构

架构参数(M)GMAC输入大小预训练权重
xcit_large_24_p8_224188.93141.232242
xcit_large_24_p8_384188.93415.003841
xcit_large_24_p16_224189.1035.862242
xcit_large_24_p16_384189.10105.353841
xcit_medium_24_p8_22484.3263.532242
xcit_medium_24_p8_38484.32186.673841
xcit_medium_24_p16_22484.4016.132242
xcit_medium_24_p16_38484.4047.393841
xcit_nano_12_p8_2243.052.162242
xcit_nano_12_p8_3843.056.343841
xcit_nano_12_p16_2243.050.562242
xcit_nano_12_p16_3843.051.643841
xcit_small_12_p8_22426.2118.692242
xcit_small_12_p8_38426.2154.923841
xcit_small_12_p16_22426.254.822242
xcit_small_12_p16_38426.2514.143841
xcit_small_24_p8_22447.6335.812242
xcit_small_24_p8_38447.63105.243841
xcit_small_24_p16_22447.679.102242
xcit_small_24_p16_38447.6726.723841
xcit_tiny_12_p8_2246.714.812242
xcit_tiny_12_p8_3846.7114.133841
xcit_tiny_12_p16_2246.721.242242
xcit_tiny_12_p16_3846.723.643841
xcit_tiny_24_p8_22412.119.212242
xcit_tiny_24_p8_38412.1127.053841
xcit_tiny_24_p16_22412.122.342242
xcit_tiny_24_p16_38412.126.873841

GMACs 和参数量对应于每种架构的默认输入大小和分类器,数据来自基准测试结果。

预训练权重

Top-1 是 results-imagenet.csv 中在所列评估规模下的 ImageNet-1k 验证准确率,- 表示该权重没有 ImageNet-1k 结果(例如预训练权重或图文权重)。Classes 是预训练分类器的大小,对于没有分类器的权重则为 none。

权重输入大小Top-1(%)评估大小类许可证Hub
xcit_large_24_p8_224.fb_dist_in1k22485.3962241000apache-2.0timm
xcit_large_24_p8_224.fb_in1k22484.3802241000apache-2.0timm
xcit_large_24_p8_384.fb_dist_in1k38486.0243841000apache-2.0timm
xcit_large_24_p16_224.fb_dist_in1k22484.9362241000apache-2.0timm
xcit_large_24_p16_224.fb_in1k22482.8862241000apache-2.0timm
xcit_large_24_p16_384.fb_dist_in1k38485.7783841000apache-2.0timm
xcit_medium_24_p8_224.fb_dist_in1k22485.0782241000apache-2.0timm
xcit_medium_24_p8_224.fb_in1k22483.7442241000apache-2.0timm
xcit_medium_24_p8_384.fb_dist_in1k38485.8423841000apache-2.0timm
xcit_medium_24_p16_224.fb_dist_in1k22484.2922241000apache-2.0timm
xcit_medium_24_p16_224.fb_in1k22482.6302241000apache-2.0timm
xcit_medium_24_p16_384.fb_dist_in1k38485.4343841000apache-2.0timm
xcit_nano_12_p8_224.fb_dist_in1k22476.3302241000apache-2.0timm
xcit_nano_12_p8_224.fb_in1k22473.9002241000apache-2.0timm
xcit_nano_12_p8_384.fb_dist_in1k38477.8103841000apache-2.0timm
xcit_nano_12_p16_224.fb_dist_in1k22472.3182241000apache-2.0timm
xcit_nano_12_p16_224.fb_in1k22469.9822241000apache-2.0timm
xcit_nano_12_p16_384.fb_dist_in1k38475.4603841000apache-2.0timm
xcit_small_12_p8_224.fb_dist_in1k22484.2522241000apache-2.0timm
xcit_small_12_p8_224.fb_in1k22483.3542241000apache-2.0timm
xcit_small_12_p8_384.fb_dist_in1k38485.0683841000apache-2.0timm
xcit_small_12_p16_224.fb_dist_in1k22483.3542241000apache-2.0timm
xcit_small_12_p16_224.fb_in1k22481.9822241000apache-2.0timm
xcit_small_12_p16_384.fb_dist_in1k38484.7343841000apache-2.0timm
xcit_small_24_p8_224.fb_dist_in1k22484.8782241000apache-2.0timm
xcit_small_24_p8_224.fb_in1k22483.8542241000apache-2.0timm
xcit_small_24_p8_384.fb_dist_in1k38485.5803841000apache-2.0timm
xcit_small_24_p16_224.fb_dist_in1k22483.8802241000apache-2.0timm
xcit_small_24_p16_224.fb_in1k22482.5882241000apache-2.0timm
xcit_small_24_p16_384.fb_dist_in1k38485.1023841000apache-2.0timm
xcit_tiny_12_p8_224.fb_dist_in1k22481.2142241000apache-2.0timm
xcit_tiny_12_p8_224.fb_in1k22479.7102241000apache-2.0timm
xcit_tiny_12_p8_384.fb_dist_in1k38482.3943841000apache-2.0timm
xcit_tiny_12_p16_224.fb_dist_in1k22478.5722241000apache-2.0timm
xcit_tiny_12_p16_224.fb_in1k22477.1222241000apache-2.0timm
xcit_tiny_12_p16_384.fb_dist_in1k38480.9543841000apache-2.0timm
xcit_tiny_24_p8_224.fb_dist_in1k22482.5782241000apache-2.0timm
xcit_tiny_24_p8_224.fb_in1k22481.8882241000apache-2.0timm
xcit_tiny_24_p8_384.fb_dist_in1k38483.7843841000apache-2.0timm
xcit_tiny_24_p16_224.fb_dist_in1k22480.4622241000apache-2.0timm
xcit_tiny_24_p16_224.fb_in1k22479.4742241000apache-2.0timm
xcit_tiny_24_p16_384.fb_dist_in1k38482.5803841000apache-2.0timm
Update on GitHub