ONNX加载
import onnx
net = onnx.load("./resnet.onnx")
onnx的__init__.py中调用了protobuf, 出现以下错误
[libprotobuf ERROR google/protobuf/io/coded_stream.cc:207] A protocol message was rejected because it was too big (more than 67108864 bytes).
解决方案
export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
参考:
libprotobuf: A protocol message was rejected because it was too big (more than 67108864 bytes) 解决方法