LoadConfigDataAndLabelMap
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说明。 |
GetModelTensorsShape
函数功能
保护成员函数，该接口由Init()调用，用于获取模型的输入输出张量形状。
函数原型
APP_ERROR GetModelTensorsShape(MxBase::ModelDesc modelDesc);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| modelDesc | 输入 | 模型基本信息结构体。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
CheckModelCompatibility
函数功能
保护成员函数，该接口由Init()调用，用于对所读取模型的输出形状进行校验。
函数原型
APP_ERROR CheckModelCompatibility();
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
类说明
该类继承了ModelPostProcessorBase用于目标检测推理后处理，用户可以根据需要继承该类并选择性实现Init、DeInit、Process等虚函数。
在线提单