From 3473ae8cf34c755c3bb9fe61ef8f9c4e413abb9b Mon Sep 17 00:00:00 2001
From: haoyuying <18844182690@163.com>
Date: Fri, 9 Oct 2020 21:00:53 +0800
Subject: [PATCH] revise download address for openpose

---
 .../keypoint_detection/openpose_body_estimation/module.py   | 6 ++----
 .../keypoint_detection/openpose_hands_estimation/module.py  | 6 ++----
 2 files changed, 4 insertions(+), 8 deletions(-)

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 dbed9ba0..011506d2 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 c86b81a1..d3fb57f0 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")
-- 
GitLab