Method createPlugin

plugin/proposalPlugin/proposalPlugin.cpp:678–779

}

IPluginV2Ext* ProposalPluginCreator::createPlugin(char const* name, PluginFieldCollection const* fc) noexcept
{
    try
    {
        PluginField const* fields = fc->fields;
        int32_t nbFields = fc->nbFields;
        int32_t inputHeight = 0;
        int32_t inputWidth = 0;
        int32_t rpnStride = 0;
        int32_t preNmsTopN = 0;
        int32_t postNmsTopN = 0;
        float roiMinSize = 0.0F;
        float nmsIouThreshold = 0.0F;
        std::vector<float> anchorSizes;
        std::vector<float> anchorRatios;

        for (int32_t i = 0; i < nbFields; ++i)
        {
            char const* attrName = fields[i].name;

            if (!strcmp(attrName, "input_height"))
            {
                PLUGIN_VALIDATE(fields[i].type == PluginFieldType::kINT32);
                inputHeight = *(static_cast<int32_t const*>(fields[i].data));
            }
            else if (!strcmp(attrName, "input_width"))
            {
                PLUGIN_VALIDATE(fields[i].type == PluginFieldType::kINT32);
                inputWidth = *(static_cast<int32_t const*>(fields[i].data));
            }
            else if (!strcmp(attrName, "rpn_stride"))
            {
                PLUGIN_VALIDATE(fields[i].type == PluginFieldType::kINT32);
                rpnStride = *(static_cast<int32_t const*>(fields[i].data));
            }
            else if (!strcmp(attrName, "roi_min_size"))
            {
                PLUGIN_VALIDATE(fields[i].type == PluginFieldType::kFLOAT32);
                roiMinSize = *(static_cast<float const*>(fields[i].data));
            }
            else if (!strcmp(attrName, "nms_iou_threshold"))
            {
                PLUGIN_VALIDATE(fields[i].type == PluginFieldType::kFLOAT32);
                nmsIouThreshold = *(static_cast<float const*>(fields[i].data));
            }
            else if (!strcmp(attrName, "pre_nms_top_n"))
            {
                PLUGIN_VALIDATE(fields[i].type == PluginFieldType::kINT32);
                preNmsTopN = *(static_cast<int32_t const*>(fields[i].data));
            }
            else if (!strcmp(attrName, "post_nms_top_n"))
            {
                PLUGIN_VALIDATE(fields[i].type == PluginFieldType::kINT32);
                postNmsTopN = *(static_cast<int32_t const*>(fields[i].data));
            }
            else if (!strcmp(attrName, "anchor_sizes"))
            {
                PLUGIN_VALIDATE(fields[i].type == PluginFieldType::kFLOAT32);