备注

您正在阅读 MMClassification 0.x 版本的文档。MMClassification 0.x 会在 2022 年末被切换为次要分支。建议您升级到 MMClassification 1.0 版本，体验更多新特性和新功能。请查阅 MMClassification 1.0 的安装教程、迁移教程以及更新日志。

mmcls.models.HRFuseScales¶

class mmcls.models.HRFuseScales(in_channels, out_channels=2048, norm_cfg={'momentum': 0.1, 'type': 'BN'}, init_cfg={'layer': 'Linear', 'std': 0.01, 'type': 'Normal'})[源代码]¶

Fuse feature map of multiple scales in HRNet.

参数

in_channels (list[int]) – The input channels of all scales.
out_channels (int) – The channels of fused feature map. Defaults to 2048.
norm_cfg (dict) – dictionary to construct norm layers. Defaults to dict(type='BN', momentum=0.1).
init_cfg (dict | list[dict], optional) – Initialization config dict. Defaults to dict(type='Normal', layer='Linear', std=0.01)).