rm::OnnxNet类 参考

ONNX-Runtime (Ort) 部署库基类 [15] 更多...

#include <rmvl/ml/ort.h>

类 rm::OnnxNet 继承关系图:

rm::OnnxNet 的协作图:

Public 成员函数
void	printModelInfo () noexcept
	打印模型信息

静态 Public 成员函数
static void	printEnvInfo () noexcept
	打印环境信息

Protected 成员函数
	OnnxNet (std::string_view model_path, OrtProvider prov)
	创建 OnnxNet 对象
std::vector< Ort::Value >	run (const std::vector< Ort::Value > &input_tensors)
	执行 ONNX Runtime 推理

Protected 属性
Ort::MemoryInfo	_memory_info
	内存分配信息
Ort::Env	_env
	环境配置
Ort::SessionOptions	_session_options
	会话选项
std::unique_ptr< Ort::Session >	_session
	会话
std::vector< const char * >	_inames
	输入名称
std::vector< const char * >	_onames
	输出名称

详细描述

ONNX-Runtime (Ort) 部署库基类 [15]

构造及析构函数说明

OnnxNet()

rm::OnnxNet::OnnxNet ( std::string_view model_path, OrtProvider prov )

protected

Python:
	rm.OnnxNet(	model_path, prov	) ->	<OnnxNet object>

创建 OnnxNet 对象

参数

[in]	model_path	模型路径，如果该路径不存在，则程序将因错误而退出
[in]	prov	Ort 提供者

函数调用图:

成员函数说明

printEnvInfo()

void rm::OnnxNet::printEnvInfo ( )

staticnoexcept

Python:
	rm.OnnxNet.printEnvInfo(		) ->	None

打印环境信息

printModelInfo()

void rm::OnnxNet::printModelInfo ( )

noexcept

Python:
	rm.OnnxNet.printModelInfo(		) ->	None

打印模型信息

函数调用图:

run()

std::vector< Ort::Value > rm::OnnxNet::run ( const std::vector< Ort::Value > & input_tensors )

protected

执行 ONNX Runtime 推理

参数

[in]

input_tensors

模型的输入 Tensors

返回

模型的输出 Tensors

函数调用图:

类成员变量说明

_env

Ort::Env rm::OnnxNet::_env

protected

环境配置

_inames

std::vector<const char *> rm::OnnxNet::_inames

protected

输入名称

_memory_info

Ort::MemoryInfo rm::OnnxNet::_memory_info

protected

内存分配信息

_onames

std::vector<const char *> rm::OnnxNet::_onames

protected

输出名称

_session

std::unique_ptr<Ort::Session> rm::OnnxNet::_session

protected

会话

_session_options

Ort::SessionOptions rm::OnnxNet::_session_options

protected

会话选项