diff --git a/x2paddle/convert.py b/x2paddle/convert.py index 75a562159edfc868819c1e4cbe1618a9304a94b9..711c974325e9001764bff7362472ba57431f9f40 100644 --- a/x2paddle/convert.py +++ b/x2paddle/convert.py @@ -163,13 +163,12 @@ def onnx2paddle(model_path, save_dir, params_merge=False): return print("Now translating model from onnx to paddle.") + from x2paddle.op_mapper.onnx_op_mapper import ONNXOpMapper from x2paddle.decoder.onnx_decoder import ONNXDecoder - model = ONNXDecoder(model_path) + from x2paddle.optimizer.onnx_optimizer import ONNXOptimizer - from x2paddle.op_mapper.onnx_op_mapper import ONNXOpMapper + model = ONNXDecoder(model_path) mapper = ONNXOpMapper(model, save_dir) - - from x2paddle.optimizer.onnx_optimizer import ONNXOptimizer optimizer = ONNXOptimizer(mapper) optimizer.delete_redundance_code() diff --git a/x2paddle/op_mapper/onnx_op_mapper.py b/x2paddle/op_mapper/onnx_op_mapper.py index 89b216d6da7b6f899a9b57d9e5f5b721c1b083c9..cbb1a7f9732fdb47bcada62d1b7a15e5fc7e4a24 100644 --- a/x2paddle/op_mapper/onnx_op_mapper.py +++ b/x2paddle/op_mapper/onnx_op_mapper.py @@ -141,7 +141,12 @@ class ONNXOpMapper(OpMapper): onnx.save(model, os.path.join(self.tmp_data_dir, 'onnx_model_infer.onnx')) - os.system('onnx_infer --save_dir=' + self.tmp_data_dir) + is_success = os.system('onnx_infer --save_dir=' + self.tmp_data_dir) + if is_success != 0: + raise Exception("onnxruntime inference onnx model failed, Please \ + confirm the correctness of onnx model by onnxruntime, \ + if onnx model is valid, you can submit issue in github." + ) return def get_dynamic_shape(self, layer):