解决:ValueError: Unsupported ONNX opset version: 15

解决在使用segment-anything项目,转onnx模型时的ValueError错误

在使用segment-anything时,想将pth模型转成onnx方便使用,但是总是报错,类似下面这种:
“File “/root/anaconda3/envs/s_samLabel/lib/python3.8/site-packages/torch/onnx/symbolic_helper.py”, line 748, in _set_opset_version
raise ValueError("Unsupported ONNX opset version: " + str(opset_version))
ValueError: Unsupported ONNX opset version: 17

整了很久,应该还是版本的问题,用的python3.8+pytorch1.8+onnx==1.14.1+onnxruntime==1.16.1始终会报这个错误。

后来的解决办法是:
使用 python3.9.16 + torch==2.0.1 + onnx==1.14.1、onnxruntime==1.16.1
然后就转换成功了,所以用比较新的项目还是用比较新的版本吧。