rm::OnnxRT类 参考

ONNX-Runtime (Ort) 部署库 [6] 更多...

#include <rmvl/ml/ort.h>

rm::OnnxRT 的协作图:

Public 成员函数
	OnnxRT (std::string_view model_path)
	创建 OnnxRT 对象 更多...
	~OnnxRT ()
void	printModelInfo ()
std::vector< size_t >	inference (const std::vector< cv::Mat > &images, const std::vector< float > &means, const std::vector< float > &stds)
	预处理，推理和后处理 更多...

详细描述

ONNX-Runtime (Ort) 部署库 [6]

构造及析构函数说明

OnnxRT()

rm::OnnxRT::OnnxRT

(

std::string_view

model_path

)

创建 OnnxRT 对象

参数

[in]

model_path

模型路径，如果该路径不存在，则程序将因错误而退出

~OnnxRT()

rm::OnnxRT::~OnnxRT

(

)

成员函数说明

inference()

std::vector<size_t> rm::OnnxRT::inference	(	const std::vector< cv::Mat > &	images,
		const std::vector< float > &	means,
		const std::vector< float > &	stds
	)

预处理，推理和后处理

参数

[in]	images	所有的输入图像
[in]	means	网络模型各通道的均值
[in]	stds	网络模型各通道的标准差

返回

与概率最高的值对应的索引向量

printModelInfo()

void rm::OnnxRT::printModelInfo

(

)

---

该类的文档由以下文件生成:

• rmvl/ml/ort.h