【Atlas200】继承MxBase类自制插件（C++），以官方例程为例

颢师傅

已于 2023-06-01 11:32:11 修改

阅读量668

点赞数

分类专栏： atlas c++ 边缘盒子文章标签： c++ 人工智能开发语言

于 2023-06-01 11:31:42 首次发布

本文链接：https://blog.csdn.net/hh1357102/article/details/130981264

版权

本文介绍了MindX SDK，一个针对昇腾AI芯片的开发工具包，用于模型部署和实时推理。重点讲解了如何继承MxBase类创建自定义插件，包括Init初始化、DeInit资源释放、Process插件入口、DefineProperties和DefineInputPorts接口的使用，以实现高性能的AI应用。

摘要由CSDN通过智能技术生成

MindX SDK简介

MindX是一款针对昇腾系列AI芯片的软件开发工具包（SDK），它提供了一系列的API和工具，帮助开发者对昇腾系列AI芯片进行开发和优化。提供了一系列的部署工具，可以将优化后的AI模型部署到昇腾AI芯片上，进行实时推理，实现低延迟、高性能的AI应用。其突出特点是使用流程编排进行开发。

总的来说，MindX SDK为开发者提供了一套完整的开发环境和工具，帮助开发者轻松地进行昇腾系列AI芯片的开发、优化和部署，实现高性能、低功耗的AI应用。
在这里插入图片描述

插件的Buffer与Metadata

Plugin表示业务流程中的基础模块，通过Element的串接构建成一个Stream。Buffer用于内部挂载解码前后的视频、图像数据，是Element之间传递的数据结构，同时也允许用户挂载元数据（Metadata），用于存放结构化数据（如目标检测结果）或过程数据（如缩放后的图像）。如下图，Metadata依附于buffer。
在这里插入图片描述

自定义插件

Init初始化接口

virtual APP_ERROR Init(std::map<std::string, std::shared_ptr<void>>& configParamMap) = 0;

插件json版：该插件名为mxpi_webpushstream0

"mxpi_webpushstream0": {
   
    "props": {
   
        "dataSourceCoordinate": "mxpi_objectpostprocessor0",
        "channelId": "0",
        "deviceId": "0"
    },
    "factory": "mxpi_webpushstream",
    "next": "fakesink0"
},

获取基础资源，获取控件的基本属性（“props”），读取的原始属性都是字符串，需要进行转换，也可以读取本地自定义的配置文件：

APP_ERROR WebPushStream::Init(std::map<std::string, std::shared_ptr<void>>& configParamMap)
{
   
     for (const auto& kv : configParamMap) {
   
        const std::string& key = kv.first;
        std::cout<<key;
     }
    channelId_ = *std::static_pointer_cast<uint>(configParamMap["channelId"]);
    if (channelId_ == -1) {
   
        LogInfo << "Failed to read the channelId";
        return APP_ERR_COMM_INIT_FAIL;
    }
    coordinateDataSource_ = *std::