timm documentation

ResNetV2 (BiT)

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.0.30).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

ResNetV2 (BiT)

类别: CNN · 代码: timm/models/resnetv2.py · 架构: 20 · 预训练权重: 24(其中 18 个包含 ImageNet-1k 结果)

采用组归一化和权重标准化的预激活 ResNet(v2),即 Big Transfer(BiT)所使用的架构,并附带原始的 BiT 权重。同时支持作为混合 Vision Transformer 骨干网络使用的非预激活变体。

论文

用法

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

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

模型 API 支持

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

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

架构

架构参数(M)GMAC输入大小预训练权重
resnetv2_1811.691.822241
resnetv2_18d11.712.062241
resnetv2_3421.803.672241
resnetv2_34d21.823.912242
resnetv2_5025.554.112241
resnetv2_50d25.574.352240
resnetv2_50d_evos25.594.332241
resnetv2_50d_frn25.594.332240
resnetv2_50d_gn25.574.382241
resnetv2_50t25.574.322240
resnetv2_50x1_bit25.554.232243
resnetv2_50x3_bit217.32145.704482
resnetv2_10144.547.832241
resnetv2_101d44.568.072240
resnetv2_101x1_bit44.5431.654482
resnetv2_101x3_bit387.93280.334482
resnetv2_15260.1911.552240
resnetv2_152d60.2011.802240
resnetv2_152x2_bit236.3446.952244
resnetv2_152x4_bit936.53844.844802

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

预训练权重

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

权重按来源分组:预训练配置中给出的论文或来源,否则为 timm 以外的 Hub 组织,否则为预训练标签的第一部分(architecture.tag)。仅包含单个权重的来源列在 other 下。

goog 标签

权重输入大小Top-1(%)评估大小类许可证Hub
resnetv2_50x1_bit.goog_distilled_in1k22482.8202241000apache-2.0timm
resnetv2_50x1_bit.goog_in21k224--21843apache-2.0timm
resnetv2_50x1_bit.goog_in21k_ft_in1k44880.3864481000apache-2.0timm
resnetv2_50x3_bit.goog_in21k224--21843apache-2.0timm
resnetv2_50x3_bit.goog_in21k_ft_in1k44883.9824481000apache-2.0timm
resnetv2_101x1_bit.goog_in21k224--21843apache-2.0timm
resnetv2_101x1_bit.goog_in21k_ft_in1k44882.3324481000apache-2.0timm
resnetv2_101x3_bit.goog_in21k224--21843apache-2.0timm
resnetv2_101x3_bit.goog_in21k_ft_in1k44884.4304481000apache-2.0timm
resnetv2_152x2_bit.goog_in21k224--21843apache-2.0timm
resnetv2_152x2_bit.goog_in21k_ft_in1k44884.4684481000apache-2.0timm
resnetv2_152x2_bit.goog_teacher_in21k_ft_in1k22482.9182241000apache-2.0timm
resnetv2_152x2_bit.goog_teacher_in21k_ft_in1k_38438483.8583841000apache-2.0timm
resnetv2_152x4_bit.goog_in21k224--21843apache-2.0timm
resnetv2_152x4_bit.goog_in21k_ft_in1k48084.9364801000apache-2.0timm

ra4 标签

权重输入大小Top-1(%)评估大小类许可证Hub
resnetv2_18.ra4_e3600_r224_in1k224(测试 288)75.3642881000apache-2.0timm
resnetv2_18d.ra4_e3600_r224_in1k224(测试 288)76.0582881000apache-2.0timm
resnetv2_34.ra4_e3600_r224_in1k224(测试 288)79.0702881000apache-2.0timm
resnetv2_34d.ra4_e3600_r224_in1k224(测试 288)79.6142881000apache-2.0timm
resnetv2_34d.ra4_e3600_r384_in1k384(测试 448)80.4304481000apache-2.0timm

a1h 标签

权重输入大小Top-1(%)评估大小类许可证Hub
resnetv2_50.a1h_in1k224(测试 288)81.4062881000apache-2.0timm
resnetv2_101.a1h_in1k224(测试 288)83.0022881000apache-2.0timm

ah 标签

权重输入大小Top-1(%)评估大小类许可证Hub
resnetv2_50d_evos.ah_in1k224(测试 288)82.0242881000apache-2.0timm
resnetv2_50d_gn.ah_in1k224(测试 288)81.9602881000apache-2.0timm
Update on GitHub