diff --git a/deploy/python/keypoint_infer.py b/deploy/python/keypoint_infer.py index ec36a193027b0cb168d85474bb8ecc0c18a5a263..e16ddd647cf58a58bb9b4c8cb239fd9e3d472673 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'))