timm documentation

BYOANet(BoTNet、HaloNet、LambdaNet)

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).
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