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说明。 |
GetLabelName
函数功能
根据索引获取类名称。
函数原型
std::string GetLabelName(int index);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| index | 输入 | 类索引。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| std::string | 类名称。 |
LoadConfigDataAndLabelMap
函数功能
保护成员函数，该接口由Init()调用，用于实现配置参数的获取。
函数原型
APP_ERROR LoadConfigDataAndLabelMap(const std::string& configPath, const std::string& labelPath);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| configPath | 输入 | 配置文件路径。 |
| labelPath | 输入 | 标签文件路径。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
在线提单