| 409 | // Standard NMS Plugin Operation |
| 410 | |
| 411 | EfficientNMSPluginCreator::EfficientNMSPluginCreator() |
| 412 | : mParam{} |
| 413 | { |
| 414 | mPluginAttributes.clear(); |
| 415 | mPluginAttributes.emplace_back(PluginField("score_threshold", nullptr, PluginFieldType::kFLOAT32, 1)); |
| 416 | mPluginAttributes.emplace_back(PluginField("iou_threshold", nullptr, PluginFieldType::kFLOAT32, 1)); |
| 417 | mPluginAttributes.emplace_back(PluginField("max_output_boxes", nullptr, PluginFieldType::kINT32, 1)); |
| 418 | mPluginAttributes.emplace_back(PluginField("background_class", nullptr, PluginFieldType::kINT32, 1)); |
| 419 | mPluginAttributes.emplace_back(PluginField("score_activation", nullptr, PluginFieldType::kINT32, 1)); |
| 420 | mPluginAttributes.emplace_back(PluginField("class_agnostic", nullptr, PluginFieldType::kINT32, 1)); |
| 421 | mPluginAttributes.emplace_back(PluginField("box_coding", nullptr, PluginFieldType::kINT32, 1)); |
| 422 | mFC.nbFields = mPluginAttributes.size(); |
| 423 | mFC.fields = mPluginAttributes.data(); |
| 424 | } |
| 425 | |
| 426 | char const* EfficientNMSPluginCreator::getPluginName() const noexcept |
| 427 | { |
nothing calls this directly
no test coverage detected