未验证 提交 fca31c81 编写于 作者: J JYChen 提交者: GitHub

fix python inference error when use fuse_norm keypoint model (#5300)

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