C++ onnxruntime从12升级到14遇到的问题

最新推荐文章于 2024-06-19 09:20:32 发布

Peter_ch_26

最新推荐文章于 2024-06-19 09:20:32 发布

阅读量1.7k

点赞数 5

文章标签： c++ onnx onnxruntime

本文链接：https://blog.csdn.net/c654528593/article/details/129840187

版权

问题

把libonnxruntime从12升级到14，遇到了一个编译问题GetInputName和GetOutputName找不到了

解决方法

原因是1.14中把 GetInputName改成了 GetInputNameAllocated, 返回值从char* 变成了AllocatedStringPtr（using Ort::AllocatedStringPtr = typedef std::unique_ptr<char, detail::AllocatedFree>
）。

std::vector<const char*> inputNodeNames; //
	std::vector<AllocatedStringPtr> inputNodeNameAllocatedStrings; // <-- newly added
	...

	auto inputNodesNum = session->GetInputCount();
	for (int i = 0; i < inputNodesNum; i++) {
		auto input_name = session->GetInputNameAllocated(i, allocator);
		inputNodeNameAllocatedStrings.push_back(std::move(input_name));
		inputNodeNames.push_back(inputNodeNameAllocatedStrings.back().get());
}