Initial snapshot: hand motion pipeline (Dyn-HaMR + dex-retargeting + SPIDER)
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,44 @@
|
||||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
from mmcv.cnn import MODELS as MMCV_MODELS
|
||||
from mmcv.cnn import build_model_from_cfg
|
||||
from mmcv.utils import Registry
|
||||
|
||||
MODELS = Registry(
|
||||
'models', build_func=build_model_from_cfg, parent=MMCV_MODELS)
|
||||
|
||||
BACKBONES = MODELS
|
||||
NECKS = MODELS
|
||||
HEADS = MODELS
|
||||
LOSSES = MODELS
|
||||
POSENETS = MODELS
|
||||
MESH_MODELS = MODELS
|
||||
|
||||
|
||||
def build_backbone(cfg):
|
||||
"""Build backbone."""
|
||||
return BACKBONES.build(cfg)
|
||||
|
||||
|
||||
def build_neck(cfg):
|
||||
"""Build neck."""
|
||||
return NECKS.build(cfg)
|
||||
|
||||
|
||||
def build_head(cfg):
|
||||
"""Build head."""
|
||||
return HEADS.build(cfg)
|
||||
|
||||
|
||||
def build_loss(cfg):
|
||||
"""Build loss."""
|
||||
return LOSSES.build(cfg)
|
||||
|
||||
|
||||
def build_posenet(cfg):
|
||||
"""Build posenet."""
|
||||
return POSENETS.build(cfg)
|
||||
|
||||
|
||||
def build_mesh_model(cfg):
|
||||
"""Build mesh model."""
|
||||
return MESH_MODELS.build(cfg)
|
||||
Reference in New Issue
Block a user