rename to PartAffinityField

open-mmlab · May 26, 2021 · ebcf6ee · ebcf6ee
1 parent c1b888c
commit ebcf6ee
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/mmpose/models/detectors/__init__.py b/mmpose/models/detectors/__init__.py
@@ -1,7 +1,11 @@
 from .bottom_up import BottomUp
 from .mesh import ParametricMesh
 from .multi_task import MultiTask
+from .paf import PartAffinityField
 from .pose_lifter import PoseLifter
 from .top_down import TopDown
 
-__all__ = ['TopDown', 'BottomUp', 'ParametricMesh', 'MultiTask', 'PoseLifter']
+__all__ = [
+    'TopDown', 'BottomUp', 'ParametricMesh', 'MultiTask', 'PoseLifter',
+    'PartAffinityField'
+]
diff --git a/mmpose/models/detectors/paf.py b/mmpose/models/detectors/paf.py
@@ -17,7 +17,7 @@
 
 
 @POSENETS.register_module()
-class BottomUp(BasePose):
+class PartAffinityField(BasePose):
     """Bottom-up pose detectors.
 
     Args: