
HRNet图像分类:基于ImageNet的HRNet模型训练
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本研究采用HRNet架构在ImageNet数据集上进行大规模预训练,旨在提升图像分类任务中的准确性与效率。
用于图像分类的高分辨率网络(HRNets)在2021年1月20日添加了一些更强大的ImageNet预训练模型,例如HRNet_W48_C_ssld_pretrained.pth,其top-1准确率达到了83.6%。在2020年3月13日,我们的论文被TPAMI接受。根据要求,我们提供了两种小型的HRNet模型,这些模型的参数和GFLOP与ResNet18相似。这两个小型模型的细分结果也可获取。
我们使用了分类头来扩展HRNet,并首先将四分辨率特征图输入瓶颈层,然后分别增加输出通道的数量至128、256、512以及1024。接着通过一个由两个步长为3x3卷积组成的下采样过程(每个都产生256个输出通道)对高分辨率表示进行处理,并将其添加到第二个高分辨率表示中。我们重复此操作两次,以细化网络结构。
这是HRNet的官方代码实现。
全部评论 (0)
还没有任何评论哟~


