关于如何修改HRNet的backbone

open-mmlab / mmpose

OpenMMLab Pose Estimation Toolbox and Benchmark.

https://mmpose.readthedocs.io/en/latest/

Apache License 2.0

5.68k stars 1.23k forks source link

关于如何修改HRNet的backbone #1435

Closed ShaneCan closed 2 years ago

ShaneCan commented 2 years ago

由于网络轻量化的需求，想要删除hrnet中的第四阶段即stage4。在config文件中直接将backbone中的stage4删除后运行训练train.py，报错： File "F:\Anaconda\envs\mmpose\lib\site-packages\mmcv\utils\registry.py", line 55, in build_from_cfg raise type(e)(f'{obj_cls.name}: {e}') KeyError: 'TopDown: "HRNet: \'stage4\'"'

请问应当如何正确修改网络结构呢？感谢回答

mm-assistant[bot] commented 2 years ago

We recommend using English or English & Chinese for issues so that we could have broader discussion.

jin-s13 commented 2 years ago

只改config是不够的，得去照着源码来改。 https://github.com/open-mmlab/mmpose/blob/16027f633431491b299cac5d2350f8753b0dfdce/mmpose/models/backbones/hrnet.py#L366