move misc.py from mmpose.utils to mmpose.model.utils

open-mmlab · Jun 3, 2022 · 811450c · 811450c
1 parent 2d19d9f
commit 811450c
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/mmpose/models/detectors/multiview_pose.py b/mmpose/models/detectors/multiview_pose.py
@@ -13,9 +13,9 @@
 from mmpose.core.camera import SimpleCamera, SimpleCameraTorch
 from mmpose.core.post_processing.post_transforms import (
     affine_transform_torch, get_affine_transform)
-from mmpose.utils.misc import torch_meshgrid_ij
 from .. import builder
 from ..builder import POSENETS
+from ..utils.misc import torch_meshgrid_ij
 from .base import BasePose
 
 

diff --git a/mmpose/models/utils/__init__.py b/mmpose/models/utils/__init__.py
@@ -1,9 +1,15 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+from ops import resize
+from realnvp import RealNVP
+
 from .ckpt_convert import pvt_convert
+from .geometry import batch_rodrigues, quat_to_rotmat, rot6d_to_rotmat
+from .misc import torch_meshgrid_ij
 from .smpl import SMPL
 from .transformer import PatchEmbed, PatchMerging, nchw_to_nlc, nlc_to_nchw
 
 __all__ = [
     'SMPL', 'PatchEmbed', 'nchw_to_nlc', 'nlc_to_nchw', 'pvt_convert',
-    'PatchMerging'
+    'PatchMerging', 'batch_rodrigues', 'quat_to_rotmat', 'rot6d_to_rotmat',
+    'resize', 'RealNVP', 'torch_meshgrid_ij'
 ]
diff --git a/mmpose/utils/misc.py → mmpose/models/utils/misc.py b/mmpose/utils/misc.py → mmpose/models/utils/misc.py