类说明
Init
函数功能
用于完成模型后处理初始化。
函数原型
APP_ERROR Init(const std::string& configPath, const std::string& labelPath, MxBase::ModelDesc modelDesc) override;
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| configPath | 输入 | 配置文件路径。 |
| labelPath | 输入 | 标签文件路径。 |
| modelDesc | 输出 | 模型描述信息。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
DeInit
函数功能
用于模型后处理的去初始化，完成资源释放。
函数原型
APP_ERROR DeInit();
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
Process
函数功能
用于推理后处理，并获取类型相关信息。
函数原型
APP_ERROR Process(std::vector<std::shared_ptr<void>>& featLayerData, int& argmaxIndex, float& confidence);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| featLayerData | 输入 | 特征数据数组。 |
| argmaxIndex | 输出 | 最可能类的索引。 |
| confidence | 输出 | 最可能类的置信度。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
类说明
该类用于实现推理后处理，用户需要继承该类并实现Init、DeInit、Process等虚函数。
Init
函数功能
用于完成模型后处理初始化。
函数原型
virtual APP_ERROR Init(const std::string& configPath, const std::string& labelPath, MxBase::ModelDesc modelDesc) = 0;
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| configPath | 输入 | 配置文件路径。 |
| labelPath | 输入 | 标签文件路径。 |
| modelDesc | 输出 | 模型描述信息。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
DeInit
函数功能
用于模型后处理的去初始化，完成资源释放。
函数原型
virtual APP_ERROR DeInit() = 0;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
Process
函数功能
用于推理后处理，获取目标、类型等信息。
函数原型
virtual APP_ERROR Process(std::vector<std::shared_ptr<void>>& featLayerData);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| featLayerData | 输入 | 特征数据数组。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
MemoryDataToHost
函数功能
把数据从Device侧拷贝到Host侧用于后处理。
函数原型
APP_ERROR MemoryDataToHost(const int index, const std::vector<std::vector<MxBase::BaseTensor>>& tensors,std::vector<std::shared_ptr<void>>& featLayerData);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| index | 输入 | 输出池索引。 |
| tensors | 输入 | 输出Tensor数组。 |
| featLayerData | 输出 | 输出特征数据数组。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
在线提单