diff --git a/hub_module/modules/image/keypoint_detection/openpose_body_estimation/module.py b/hub_module/modules/image/keypoint_detection/openpose_body_estimation/module.py index dbed9ba0b2589a826d518be5d6ac128698158530..011506d260b9b2ca01eaa5ac5bdb717044029364 100644 --- a/hub_module/modules/image/keypoint_detection/openpose_body_estimation/module.py +++ b/hub_module/modules/image/keypoint_detection/openpose_body_estimation/module.py @@ -121,11 +121,9 @@ class BodyPoseModel(nn.Layer): print("load custom checkpoint success") else: - checkpoint = os.path.join(self.directory, 'body_estimation.pdparams') + checkpoint = os.path.join(self.directory, 'openpose_body.pdparams') if not os.path.exists(checkpoint): - os.system( - 'wget https://bj.bcebos.com/paddlehub/model/image/keypoint_detection/body_estimation.pdparams -O ' + - checkpoint) + os.system('wget https://paddlehub.bj.bcebos.com/dygraph/pose/openpose_body.pdparams -O ' + checkpoint) model_dict = paddle.load(checkpoint)[0] self.set_dict(model_dict) print("load pretrained checkpoint success") diff --git a/hub_module/modules/image/keypoint_detection/openpose_hands_estimation/module.py b/hub_module/modules/image/keypoint_detection/openpose_hands_estimation/module.py index c86b81a1b2097445f0f03b8f393e0727be6d48e9..d3fb57f0d894cd86d64b56c3a66afecedb7fd721 100644 --- a/hub_module/modules/image/keypoint_detection/openpose_hands_estimation/module.py +++ b/hub_module/modules/image/keypoint_detection/openpose_hands_estimation/module.py @@ -102,11 +102,9 @@ class HandPoseModel(nn.Layer): print("load custom checkpoint success") else: - checkpoint = os.path.join(self.directory, 'hand_estimation.pdparams') + checkpoint = os.path.join(self.directory, 'openpose_hand.pdparams') if not os.path.exists(checkpoint): - os.system( - 'wget https://bj.bcebos.com/paddlehub/model/image/keypoint_detection/hand_estimation.pdparams -O ' + - checkpoint) + os.system('wget https://paddlehub.bj.bcebos.com/dygraph/pose/openpose_hand.pdparams -O ' + checkpoint) model_dict = paddle.load(checkpoint)[0] self.set_dict(model_dict) print("load pretrained checkpoint success")