页次: 1
问题找到了,是onnx模型需要精简一下,我用的是
https://github.com/daquexian/onnx-simplifier
精简后可以用libonnx推理了,谢谢xboot的建议。
参考这个程序,
https://github.com/xboot/libonnx/blob/master/examples/hello/main.c打印下看是什么情况,如果ctx是null,估计是文件不合规吧
/* * Dump onnx context */ onnx_context_dump(ctx, 0);
因为是null所以也dump不出来,怎么样的onnx文件算是合规的呢?我的onnx文件是torch.onnx导出来的
torch.onnx.export(self.model, dummy_input,
os.path.join(self.save_dir, '{}-{:.4f}-best_model.onnx'.format(epoch, best_acc)))
页次: 1