From ebcf6ee413ef4cdfdc78261054fe61cff958ba5e Mon Sep 17 00:00:00 2001 From: jinsheng Date: Wed, 26 May 2021 14:41:16 +0800 Subject: [PATCH] rename to PartAffinityField --- mmpose/models/detectors/__init__.py | 6 +++++- mmpose/models/detectors/paf.py | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/mmpose/models/detectors/__init__.py b/mmpose/models/detectors/__init__.py index 0667f84240..59b978b5bf 100644 --- 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 index 74518766f5..3ca9bba4d5 100644 --- 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: