session_.GetInputName(i,allocator);session_.GetOutputName(i,allocator); 升级到1.13.1版本之后,上面的函数没了,只有下面的函数: 代码语言:javascript 复制 session_.GetInputNameAllocated(i,allocator);session_.GetOutputNameAllocated(i,allocator); 修改之后,我依然跟之前一样把输入名称跟输出名称保存在两个全局...
session_.GetInputName(i, allocator);session_.GetOutputName(i, allocator); 升级到1.13.1版本之后,上面的函数没了,只有下面的函数: session_.GetInputNameAllocated(i, allocator);session_.GetOutputNameAllocated(i, allocator); 修改之后,我依然跟之前一样把输入名称跟输出名称保存在两个全局的std::vector里面,...
auto input_name = session_->GetInputNameAllocated(i, allocator); input_node_names.push_back(input_name.get()); auto inputShapeInfo = session_->GetInputTypeInfo(i).GetTensorTypeAndShapeInfo().GetShape(); int ch = inputShapeInfo[1]; input_h = inputShapeInfo[2]; input_w = inputShapeI...
_inputName = _OrtSession->GetInputName(0, allocator); _inputNodeNames.push_back(_inputName); #else _inputName = std::move(_OrtSession->GetInputNameAllocated(0, allocator)); _inputNodeNames.push_back(_inputName.get()); #endif //std::cout << _inputNodeNames[0] << std::endl; Ort::Ty...
int input_nodes_num = static_cast<int>(session_->GetInputCount()); int output_nodes_num = static_cast<int>(session_->GetOutputCount()); for (int i = 0; i < input_nodes_num; i++) { auto input_name = session_->GetInputNameAllocated(i, allocator); ...
size_t inputNodeCount= session.GetInputCount(); std::cout << "输入节点数量:" << inputNodeCount << "\n"; Ort::AllocatorWithDefaultOptions allocator; std::shared_ptr<char> inputName = std::move(session.GetInputNameAllocated(0, allocator)); ...
; //获取输入信息intinput_w=0; intinput_h=0; for(inti=0;i< numInputNodes; i++) { auto input_name = session_.GetInputNameAllocated(i, allocator); input_node_names.push_back(input_name.get()); Ort::TypeInfo input_type_info = session_.GetInputTypeInfo(i); auto input_tensor_info ...
4.3.1 ONNX Runtime下载 可以从https://github.com/microsoft/onnxruntime/releases下载到编译好的依赖库,这里的演示环境是Win10, IDE是vs2019,我下载v1.12.1 GPU这个版本。 4.3.2 OpenCV库 因为使用的模型需要显示出效果图,这里使用OpenCV来做图像预处理相关的操作,可以从OpenCV官网下载到所需的库: ...
Ort::AllocatedStringPtr input_name_ptr = session.GetInputNameAllocated(i, allocator); std::string input_name = std::string(input_name_ptr.get()); input_names.push_back(input_name);std::cout << "input_name: " << input_names[i] << std::endl;float...
GetInputNameAllocated(i, allocator); auto shape = session.GetInputTypeInfo(i).GetTensorTypeAndShapeInfo().GetShape(); std::cout << "Input Number: " << i << std::endl; std::cout << " Input Name: " << name.get() << std::endl; std::cout << " Input Shape: " << shape <...