open-mmlab / mmpose

OpenMMLab Pose Estimation Toolbox and Benchmark.
https://mmpose.readthedocs.io/en/latest/
Apache License 2.0
5.59k stars 1.22k forks source link

[Feature] efficientnet should be added #2379

Closed sayoko17 closed 1 year ago

sayoko17 commented 1 year ago

What is the feature?

efficientnet 在mmdetection里有运用,可以添加一个backbone将efficinetnet加入进来吗?自己尝试过但是可能因为mmdet3.0 2.0有点混乱,加过来bug很多,config里neck的部分也不知道该不该删掉,有点混乱中……能否加一个efficinetnet的backbone和config?

Any other context?

No response

Tau-J commented 1 year ago

你好,感谢使用 MMPose。 新版本的 OpenMMLab 系列算法库之间是支持跨库调用代码的,所以可以在 config 里直接使用其他 mm 算法库已经支持的各种模块,包括 backbone。下面是 RTMPose 跨库使用 mmdet 中定义的 CSPNeXt 的例子:

# model settings
model = dict(
    type='TopdownPoseEstimator',
    data_preprocessor=dict(
     ... 省略 ...
    backbone=dict(
        _scope_='mmdet',
        type='CSPNeXt',
        arch='P5',
        expand_ratio=0.5,
        deepen_factor=0.67,
        widen_factor=0.75,
        out_indices=(4, ),
        channel_attention=True,
        norm_cfg=dict(type='SyncBN'),
        act_cfg=dict(type='SiLU'),
        init_cfg=dict(
            type='Pretrained',
            prefix='backbone.',
            checkpoint='https://download.openmmlab.com/mmpose/v1/projects/'
            'rtmposev1/cspnext-m_udp-aic-coco_210e-256x192-f2f7d6f6_20230130.pth'  # noqa
        )),
        ... 省略 ...

完整的 config 可以看这里