create distiller folder

open-mmlab · Aug 29, 2023 · 5cfd51b · 5cfd51b
1 parent 084212b
commit 5cfd51b
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 7 deletions.
diff --git a/mmpose/models/__init__.py b/mmpose/models/__init__.py
@@ -4,12 +4,21 @@
                       build_head, build_loss, build_neck, build_pose_estimator,
                       build_posenet)
 from .data_preprocessors import *  # noqa
+from .distillers import *  # noqa
 from .heads import *  # noqa
 from .losses import *  # noqa
 from .necks import *  # noqa
 from .pose_estimators import *  # noqa
 
 __all__ = [
-    'BACKBONES', 'HEADS', 'NECKS', 'LOSSES', 'build_backbone', 'build_head',
-    'build_loss', 'build_posenet', 'build_neck', 'build_pose_estimator'
+    'BACKBONES',
+    'HEADS',
+    'NECKS',
+    'LOSSES',
+    'build_backbone',
+    'build_head',
+    'build_loss',
+    'build_posenet',
+    'build_neck',
+    'build_pose_estimator',
 ]
diff --git a/mmpose/models/distillers/__init__.py b/mmpose/models/distillers/__init__.py
@@ -0,0 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
+from .dwpose_distiller import DWPoseDistiller
+
+__all__ = ['DWPoseDistiller']
diff --git a/...odels/pose_estimators/dwpose_distiller.py → mmpose/models/distillers/dwpose_distiller.py b/...odels/pose_estimators/dwpose_distiller.py → mmpose/models/distillers/dwpose_distiller.py
diff --git a/mmpose/models/pose_estimators/__init__.py b/mmpose/models/pose_estimators/__init__.py
@@ -1,10 +1,6 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 from .bottomup import BottomupPoseEstimator
-from .dwpose_distiller import DWPoseDistiller
 from .pose_lifter import PoseLifter
 from .topdown import TopdownPoseEstimator
 
-__all__ = [
-    'TopdownPoseEstimator', 'BottomupPoseEstimator', 'PoseLifter',
-    'DWPoseDistiller'
-]
+__all__ = ['TopdownPoseEstimator', 'BottomupPoseEstimator', 'PoseLifter']
diff --git a/..._pose_estimators/test_dwpose_distiller.py → .../test_distillers/test_dwpose_distiller.py b/..._pose_estimators/test_dwpose_distiller.py → .../test_distillers/test_dwpose_distiller.py