timm documentation

BYOANet(BoTNet、HaloNet、LambdaNet)

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

BYOANet(BoTNet、HaloNet、LambdaNet)

类别: 混合 · 代码: timm/models/byoanet.py · 架构: 15 · 预训练权重: 13(13 个具有 ImageNet-1k 结果)

使用 ByobNet 块/阶段配置系统构建的网络,如果在 ResNet 风格的阶段中放置自注意力(或类似)层:Bottleneck Transformers (BoTNet)、HaloNets 和 Lambda ResNets。模块文档字符串将这些定义标记为实验性。

论文

用法

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

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

模型 API 支持

功能ByobNet
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) 仅支持方形尺寸,✗ 不支持,– 无需支持(该模型本身已接受任意输入尺寸)。通过在每个功能上运行随机初始化的模型(ByobNet 使用 halonet_h1,尺寸为 256)进行验证,预期同一类的其他架构表现相同。有关详细信息,请参阅 模型 API 页面。

架构

架构参数(M)GMAC输入大小预训练权重
botnet26t_25612.493.322561
botnet50ts_25622.745.542560
eca_botnext26ts_25610.592.462561
eca_halonext26ts10.762.442561
halo2botnet50ts_25622.645.022561
halonet26t12.483.192561
halonet50ts22.735.302561
halonet_h18.103.002560
haloregnetz_b11.681.972241
lambda_resnet26rpt_25610.993.162561
lambda_resnet26t10.963.022561
lambda_resnet50ts21.545.072561
lamhalobotnet50ts_25622.575.022561
sebotnet33ts_25613.703.892561
sehalonet33ts13.693.552561

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

预训练权重

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

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

a1h 标签

权重输入大小Top-1(%)评估大小类许可证Hub
halo2botnet50ts_256.a1h_in1k25682.0842561000apache-2.0timm
halonet26t.a1h_in1k25679.1342561000apache-2.0timm
halonet50ts.a1h_in1k25681.6402561000apache-2.0timm
lambda_resnet50ts.a1h_in1k25681.1842561000apache-2.0timm
lamhalobotnet50ts_256.a1h_in1k25681.5522561000apache-2.0timm
sebotnet33ts_256.a1h_in1k25681.1562561000apache-2.0timm

tags

权重输入大小Top-1(%)评估大小类许可证Hub
botnet26t_256.c1_in1k25679.2562561000apache-2.0timm
eca_botnext26ts_256.c1_in1k25679.2622561000apache-2.0timm
eca_halonext26ts.c1_in1k25679.5182561000apache-2.0timm
lambda_resnet26rpt_256.c1_in1k25678.9602561000apache-2.0timm
lambda_resnet26t.c1_in1k25679.1262561000apache-2.0timm

其他

权重输入大小Top-1(%)评估大小类许可证Hub
haloregnetz_b.ra3_in1k22481.0582241000apache-2.0timm
sehalonet33ts.ra2_in1k25680.9762561000apache-2.0timm
Update on GitHub