update module import

This commit is contained in:
Wei-Chen-hub
2024-05-30 17:06:17 +08:00
parent 5bda95bad2
commit 100d467093
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -8,7 +8,7 @@ from utils.transforms import rot6d_to_axis_angle, restore_bbox
from config import cfg
import math
import copy
from mmposesmplerx.models import build_posenet
from mmpose_smplerx.models import build_posenet
from mmengine.config import Config
class Model(nn.Module):
@@ -3,7 +3,7 @@ from mmengine.registry import MODELS as MMCV_MODELS
from mmengine import Registry
from mmengine.registry import build_from_cfg, build_model_from_cfg
MODELS = Registry('models', parent=MMCV_MODELS, locations=['mmpose.models'])
MODELS = Registry('models', parent=MMCV_MODELS, locations=['mmpose_smplerx.models'])
BACKBONES = MODELS
NECKS = MODELS