HRNet

类别： CNN · 代码： timm/models/hrnet.py · 架构： 11 · 预训练权重： 14（14 个具有 ImageNet-1k 结果）

高分辨率网络在整个网络中保留一个高分辨率分支，并并行添加多个低分辨率分支，在它们之间反复交换信息。

论文

用法

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

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

模型 API 支持

功能	`HighResolutionNet`
`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) 仅支持方形尺寸，✗ 不支持，– 无需支持（模型本身已接受任意输入尺寸）。通过在每个功能上运行随机初始化的模型（HighResolutionNet 使用 hrnet_w18_small，尺寸为 224）进行验证，预期同一类的其他架构表现相同。详情请参阅 Model API 页面。

架构

架构	参数（M）	GMAC	输入大小	预训练权重
`hrnet_w18`	21.30	4.32	224	2
`hrnet_w18_small`	13.19	1.61	224	2
`hrnet_w18_small_v2`	15.60	2.62	224	2
`hrnet_w18_ssld`	21.30	4.32	224	1
`hrnet_w30`	37.71	8.15	224	1
`hrnet_w32`	41.23	8.97	224	1
`hrnet_w40`	57.56	12.75	224	1
`hrnet_w44`	67.06	14.94	224	1
`hrnet_w48`	77.47	17.34	224	1
`hrnet_w48_ssld`	77.47	17.34	224	1
`hrnet_w64`	128.06	28.97	224	1

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

预训练权重

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

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

ms 标签

权重	输入大小	Top-1（%）	评估大小	类	许可证	Hub
`hrnet_w18.ms_aug_in1k`	224	78.110	224	1000	mit	timm
`hrnet_w18.ms_in1k`	224	76.770	224	1000	mit	timm
`hrnet_w18_small.ms_in1k`	224	72.340	224	1000	mit	timm
`hrnet_w18_small_v2.ms_in1k`	224	75.094	224	1000	mit	timm
`hrnet_w30.ms_in1k`	224	78.202	224	1000	mit	timm
`hrnet_w32.ms_in1k`	224	78.452	224	1000	mit	timm
`hrnet_w40.ms_in1k`	224	78.928	224	1000	mit	timm
`hrnet_w44.ms_in1k`	224	78.890	224	1000	mit	timm
`hrnet_w48.ms_in1k`	224	79.328	224	1000	mit	timm
`hrnet_w64.ms_in1k`	224	79.458	224	1000	mit	timm

gluon 标签

权重	输入大小	Top-1（%）	评估大小	类	许可证	Hub
`hrnet_w18_small.gluon_in1k`	224	73.934	224	1000	apache-2.0	timm
`hrnet_w18_small_v2.gluon_in1k`	224	78.202	224	1000	apache-2.0	timm

paddle 标签

权重	输入大小	Top-1（%）	评估大小	类	许可证	Hub
`hrnet_w18_ssld.paddle_in1k`	224（测试 288）	82.052	288	1000	mit	timm
`hrnet_w48_ssld.paddle_in1k`	224（测试 288）	84.484	288	1000	mit	timm

Update on GitHub