Spaces:
Running
on
L40S
Running
on
L40S
File size: 951 Bytes
d7e58f0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
# Copyright (c) OpenMMLab. All rights reserved.
from mmcv.utils import Registry
from .detr_head import DeformableDETRHead, DETRHead
from .expose_head import ExPoseBodyHead, ExPoseFaceHead, ExPoseHandHead
from .hmr_head import HMRHead
from .hybrik_head import HybrIKHead
from .pare_head import PareHead
HEADS = Registry('heads')
HEADS.register_module(name='HybrIKHead', module=HybrIKHead)
HEADS.register_module(name='HMRHead', module=HMRHead)
HEADS.register_module(name='PareHead', module=PareHead)
HEADS.register_module(name='ExPoseBodyHead', module=ExPoseBodyHead)
HEADS.register_module(name='ExPoseHandHead', module=ExPoseHandHead)
HEADS.register_module(name='ExPoseFaceHead', module=ExPoseFaceHead)
HEADS.register_module(name='DETRHead', module=DETRHead)
HEADS.register_module(name='DeformableDETRHead', module=DeformableDETRHead)
def build_head(cfg):
"""Build head."""
if cfg is None:
return None
return HEADS.build(cfg)
|