From 100d4670938e77cee90464dff2bc031a05e1c481 Mon Sep 17 00:00:00 2001
From: Wei-Chen-hub <1259566226@qq.com>
Date: Thu, 30 May 2024 17:06:17 +0800
Subject: [PATCH] update module import

---
 main/SMPLer_X.py                                        | 2 +-
 main/transformer_utils/mmpose_smplerx/models/builder.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/main/SMPLer_X.py b/main/SMPLer_X.py
index 86fad6a..57e7e03 100644
--- a/main/SMPLer_X.py
+++ b/main/SMPLer_X.py
@@ -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):
diff --git a/main/transformer_utils/mmpose_smplerx/models/builder.py b/main/transformer_utils/mmpose_smplerx/models/builder.py
index 5fa61ca..a1f909f 100644
--- a/main/transformer_utils/mmpose_smplerx/models/builder.py
+++ b/main/transformer_utils/mmpose_smplerx/models/builder.py
@@ -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