Merge 5cfd51b into 084212b

Author: Tau-J

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',
 ]

mmpose/models/distillers/__init__.py

@@ -0,0 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
+from .dwpose_distiller import DWPoseDistiller
+
+__all__ = ['DWPoseDistiller']

mmpose/models/pose_estimators/dwpose_distiller.py renamed to mmpose/models/distillers/dwpose_distiller.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']