超解像モデルの 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.
どうすればエラーを解決できますか?