ConstructProtobuf

函数功能

将模型后处理使用的数据结构转换为protobuf数据结构,写入元数据用以插件间通讯。根据入参数据类型选择重载函数。

函数原型

1
2
std::shared_ptr<MxTools::MxpiObjectList> ConstructProtobuf(
    const std::vector<std::vector<MxBase::ObjectInfo>> &objectInfos, std::string dataSource);
1
2
std::shared_ptr<MxTools::MxpiClassList> ConstructProtobuf(
    const std::vector<std::vector<MxBase::ClassInfo>>& classInfos,std::string dataSource);
1
2
std::shared_ptr<MxTools::MxpiImageMaskList> ConstructProtobuf(
    const std::vector<MxBase::SemanticSegInfo> &semanticSegInfos, std::string dataSource);
1
2
std::shared_ptr<MxTools::MxpiTextsInfoList> ConstructProtobuf(
    const std::vector<MxBase::TextsInfo>& textsInfo, std::string dataSource);
1
2
std::shared_ptr<MxTools::MxpiTextObjectList> ConstructProtobuf(
    const std::vector<std::vector<MxBase::TextObjectInfo>>& textObjectInfos, std::string dataSource);
1
2
std::shared_ptr<MxTools::MxpiPoseList> ConstructProtobuf(
    const std::vector<std::vector<MxBase::KeyPointDetectionInfo>>& keyPointInfos, std::string dataSource);

参数说明

参数名

说明

objectInfos

目标检测结果。

classInfos

分类结果。

semanticSegInfos

语义分割结果。

textsInfo

文本生成结果。

textObjectInfos

文本框检测结果。

keyPointInfos

关键点检测结果。

dataSource

上游插件的名称。

返回参数说明

数据结构

说明

std::shared_ptr<MxTools::MxpiObjectList>

与入参类型对应protobuf的智能指针。

std::shared_ptr<MxTools::MxpiClassList>

std::shared_ptr<MxTools::MxpiImageMaskList>

std::shared_ptr<MxTools::MxpiTextsInfoList>

std::shared_ptr<MxTools::MxpiTextObjectList>