提交 029ad3af 编写于 作者: W wuzewu

Fix export bug

上级 74be6033
......@@ -191,8 +191,7 @@ class MultiLangOCR:
v0, v1, v2 = paddle2onnx.__version__.split('.')
if int(v1) < 9:
raise ImportError("paddle2onnx>=0.9.0 is required")
if input_shape_dict is None:
input_shape_dict = {'x': [-1, 3, -1, -1]}
if input_shape_dict is not None and not isinstance(input_shape_dict, dict):
raise Exception("input_shape_dict should be dict, eg. {'x': [-1, 3, -1, -1]}.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册