update module import
This commit is contained in:
+1
-1
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user