From fca31c8133ab2c13504256bbb2d843050c9cb3e8 Mon Sep 17 00:00:00 2001 From: JYChen Date: Sat, 5 Mar 2022 21:49:22 +0800 Subject: [PATCH] fix python inference error when use fuse_norm keypoint model (#5300) --- deploy/python/keypoint_infer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deploy/python/keypoint_infer.py b/deploy/python/keypoint_infer.py index ec36a1930..e16ddd647 100644 --- a/deploy/python/keypoint_infer.py +++ b/deploy/python/keypoint_infer.py @@ -288,7 +288,7 @@ def create_inputs(imgs, im_info): inputs (dict): input of model """ inputs = {} - inputs['image'] = np.stack(imgs, axis=0) + inputs['image'] = np.stack(imgs, axis=0).astype('float32') im_shape = [] for e in im_info: im_shape.append(np.array((e['im_shape'])).astype('float32')) -- GitLab