From 8a685c4b67807907ce0012aba497bab1f76b31cf Mon Sep 17 00:00:00 2001 From: JYChen Date: Fri, 5 Nov 2021 09:53:28 +0800 Subject: [PATCH] update code reference information for keypoint (#4458) --- deploy/python/keypoint_preprocess.py | 5 ++++- ppdet/data/transform/keypoint_operators.py | 5 ++++- ppdet/modeling/backbones/lite_hrnet.py | 4 ++++ ppdet/modeling/keypoint_utils.py | 5 ++++- 4 files changed, 16 insertions(+), 3 deletions(-) diff --git a/deploy/python/keypoint_preprocess.py b/deploy/python/keypoint_preprocess.py index 97a747c4c..07e097b9b 100644 --- a/deploy/python/keypoint_preprocess.py +++ b/deploy/python/keypoint_preprocess.py @@ -109,7 +109,10 @@ def get_affine_transform(center, def get_warp_matrix(theta, size_input, size_dst, size_target): - """Calculate the transformation matrix under the constraint of unbiased. + """This code is based on + https://github.com/open-mmlab/mmpose/blob/master/mmpose/core/post_processing/post_transforms.py + + Calculate the transformation matrix under the constraint of unbiased. Paper ref: Huang et al. The Devil is in the Details: Delving into Unbiased Data Processing for Human Pose Estimation (CVPR 2020). diff --git a/ppdet/data/transform/keypoint_operators.py b/ppdet/data/transform/keypoint_operators.py index 81770b63e..48957faaa 100644 --- a/ppdet/data/transform/keypoint_operators.py +++ b/ppdet/data/transform/keypoint_operators.py @@ -790,7 +790,10 @@ class ToHeatmapsTopDown_DARK(object): @register_keypointop class ToHeatmapsTopDown_UDP(object): - """to generate the gaussian heatmaps of keypoint for heatmap loss. + """This code is based on: + https://github.com/HuangJunJie2017/UDP-Pose/blob/master/deep-high-resolution-net.pytorch/lib/dataset/JointsDataset.py + + to generate the gaussian heatmaps of keypoint for heatmap loss. ref: Huang et al. The Devil is in the Details: Delving into Unbiased Data Processing for Human Pose Estimation (CVPR 2020). diff --git a/ppdet/modeling/backbones/lite_hrnet.py b/ppdet/modeling/backbones/lite_hrnet.py index da98be3a4..8f0ded109 100644 --- a/ppdet/modeling/backbones/lite_hrnet.py +++ b/ppdet/modeling/backbones/lite_hrnet.py @@ -11,6 +11,10 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. +""" +This code is based on +https://github.com/HRNet/Lite-HRNet/blob/hrnet/models/backbones/litehrnet.py +""" import paddle import paddle.nn as nn diff --git a/ppdet/modeling/keypoint_utils.py b/ppdet/modeling/keypoint_utils.py index b3f84da7d..4910eb0b7 100644 --- a/ppdet/modeling/keypoint_utils.py +++ b/ppdet/modeling/keypoint_utils.py @@ -96,7 +96,10 @@ def get_affine_transform(center, def get_warp_matrix(theta, size_input, size_dst, size_target): - """Calculate the transformation matrix under the constraint of unbiased. + """This code is based on + https://github.com/open-mmlab/mmpose/blob/master/mmpose/core/post_processing/post_transforms.py + + Calculate the transformation matrix under the constraint of unbiased. Paper ref: Huang et al. The Devil is in the Details: Delving into Unbiased Data Processing for Human Pose Estimation (CVPR 2020). -- GitLab