1

超解像モデルの ONNX モデルをテストしていますが、このサンプル プログラムの実行中にエラーが発生しました。

私の ONNX バージョンは 1.5.0 で、onnxruntime は 1.4.0 です。Onnxruntime は pip を使用してインストールされました。Pytorch のバージョンは 1.6.0 です

エラーはort_session = onnxruntime.InferenceSession('/home/itc/pytorch/sub_pixel_cnn_2016/model/super-resolution-10.onnx')

onnx モデルのロード中にエラーが発生しました。

Traceback (most recent call last):
  File "test.py", line 73, in <module>
    ort_session = onnxruntime.InferenceSession('/home/itc/pytorch/sub_pixel_cnn_2016/model/super-resolution-10.onnx')
  File "/home/itc/pytorch/lib/python3.7/site-packages/onnxruntime/capi/session.py", line 158, in __init__
    self._load_model(providers or [])
  File "/home/itc/pytorch/lib/python3.7/site-packages/onnxruntime/capi/session.py", line 166, in _load_model
    True)
RuntimeError: /onnxruntime_src/onnxruntime/core/session/inference_session.cc:238 onnxruntime::InferenceSession::InferenceSession(const onnxruntime::SessionOptions&, const onnxruntime::Environment&, const string&) status.IsOK() was false. Given model could not be parsed while creating inference session. Error message: Protobuf parsing failed.
 

どうすればエラーを解決できますか?

4

1 に答える 1