未验证 提交 17f4307d 编写于 作者: W wangxinxin08 提交者: GitHub

fix create_inputs of deploy/python/infer.py (#1411)

上级 a8103906
......@@ -272,7 +272,7 @@ def create_inputs(im, im_info, model_arch='YOLO'):
if 'YOLO' in model_arch:
im_size = np.array([origin_shape]).astype('int32')
inputs['im_size'] = im_size
elif 'RetinaNet' or 'EfficientDet' in model_arch:
elif 'RetinaNet' in model_arch or 'EfficientDet' in model_arch:
scale = scale_x
im_info = np.array([resize_shape + [scale]]).astype('float32')
inputs['im_info'] = im_info
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册