提交 3473ae8c 编写于 作者: H haoyuying

revise download address for openpose

上级 6eb94743
......@@ -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")
......
......@@ -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")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册