Enable NMS in BoxesLabelsScoresConverter (#672)

* Enable NMS in BoxesLabelsScoresConverter

* Fix header

* Fix NMS and add IOU threshold to bbox converter

* Fix code style

---------

Co-authored-by: Tomasz Bujewski <tomasz.bujewski@intel.com>

Author: OskarFiedot

src/monolithic/gst/inference_elements/common/post_processor/converters/to_roi/blob_to_roi_converter.cpp

@@ -57,11 +57,14 @@ BlobToMetaConverter::Ptr BlobToROIConverter::create(BlobToMetaConverter::Initial
     else if (converter_name == DetectionOutputConverter::getName())
         return BlobToMetaConverter::Ptr(new DetectionOutputConverter(std::move(initializer), confidence_threshold));
     else if (converter_name == BoxesLabelsConverter::getName())
-        return BlobToMetaConverter::Ptr(new BoxesLabelsConverter(std::move(initializer), confidence_threshold));
+        return BlobToMetaConverter::Ptr(
+            new BoxesLabelsConverter(std::move(initializer), confidence_threshold, iou_threshold));
     else if (converter_name == BoxesScoresConverter::getName())
-        return BlobToMetaConverter::Ptr(new BoxesScoresConverter(std::move(initializer), confidence_threshold));
+        return BlobToMetaConverter::Ptr(
+            new BoxesScoresConverter(std::move(initializer), confidence_threshold, iou_threshold));
     else if (converter_name == BoxesConverter::getName())
-        return BlobToMetaConverter::Ptr(new BoxesConverter(std::move(initializer), confidence_threshold));
+        return BlobToMetaConverter::Ptr(
+            new BoxesConverter(std::move(initializer), confidence_threshold, iou_threshold));
     else if (converter_name == YOLOv2Converter::getName() || converter_name == YOLOv3Converter::getName() ||
              converter_name == YOLOv4Converter::getName() || converter_name == YOLOv5Converter::getName())
         return YOLOBaseConverter::create(std::move(initializer), converter_name, confidence_threshold);

src/monolithic/gst/inference_elements/common/post_processor/converters/to_roi/boxes.h

@@ -1,5 +1,5 @@
 /*******************************************************************************
- * Copyright (C) 2021 Intel Corporation
+ * Copyright (C) 2021-2026 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  ******************************************************************************/
@@ -30,8 +30,8 @@ class BoxesConverter : public BoxesLabelsScoresConverter {
     }
 
   public:
-    BoxesConverter(BlobToMetaConverter::Initializer initializer, double confidence_threshold)
-        : BoxesLabelsScoresConverter(std::move(initializer), confidence_threshold) {
+    BoxesConverter(BlobToMetaConverter::Initializer initializer, double confidence_threshold, double iou_threshold)
+        : BoxesLabelsScoresConverter(std::move(initializer), confidence_threshold, iou_threshold) {
     }
 
     static bool isValidModelOutputs(const std::map<std::string, std::vector<size_t>> &model_outputs_info) {

src/monolithic/gst/inference_elements/common/post_processor/converters/to_roi/boxes_labels.h

@@ -1,5 +1,5 @@
 /*******************************************************************************
- * Copyright (C) 2021-2025 Intel Corporation
+ * Copyright (C) 2021-2026 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  ******************************************************************************/
@@ -28,8 +28,9 @@ class BoxesLabelsConverter : public BoxesLabelsScoresConverter {
     std::pair<size_t, float> getLabelIdConfidence(const InferenceBackend::OutputBlob::Ptr &, size_t, float) const final;
 
   public:
-    BoxesLabelsConverter(BlobToMetaConverter::Initializer initializer, double confidence_threshold)
-        : BoxesLabelsScoresConverter(std::move(initializer), confidence_threshold) {
+    BoxesLabelsConverter(BlobToMetaConverter::Initializer initializer, double confidence_threshold,
+                         double iou_threshold)
+        : BoxesLabelsScoresConverter(std::move(initializer), confidence_threshold, iou_threshold) {
     }
 
     static bool isValidModelOutputs(const std::map<std::string, std::vector<size_t>> &model_outputs_info);

src/monolithic/gst/inference_elements/common/post_processor/converters/to_roi/boxes_labels_scores_base.h

@@ -1,5 +1,5 @@
 /*******************************************************************************
- * Copyright (C) 2021-2025 Intel Corporation
+ * Copyright (C) 2021-2026 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  ******************************************************************************/
@@ -39,8 +39,9 @@ class BoxesLabelsScoresConverter : public BlobToROIConverter {
                                                                       size_t height) const;
 
   public:
-    BoxesLabelsScoresConverter(BlobToMetaConverter::Initializer initializer, double confidence_threshold)
-        : BlobToROIConverter(std::move(initializer), confidence_threshold, false, 0.0) {
+    BoxesLabelsScoresConverter(BlobToMetaConverter::Initializer initializer, double confidence_threshold,
+                               double iou_threshold)
+        : BlobToROIConverter(std::move(initializer), confidence_threshold, true, iou_threshold) {
     }
 
     TensorsTable convert(const OutputBlobs &output_blobs) override;

src/monolithic/gst/inference_elements/common/post_processor/converters/to_roi/boxes_scores.h

@@ -1,5 +1,5 @@
 /*******************************************************************************
- * Copyright (C) 2021-2022 Intel Corporation
+ * Copyright (C) 2021-2026 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  ******************************************************************************/
@@ -31,8 +31,9 @@ class BoxesScoresConverter : public BoxesLabelsScoresConverter {
                                                               size_t height) const override final;
 
   public:
-    BoxesScoresConverter(BlobToMetaConverter::Initializer initializer, double confidence_threshold)
-        : BoxesLabelsScoresConverter(std::move(initializer), confidence_threshold) {
+    BoxesScoresConverter(BlobToMetaConverter::Initializer initializer, double confidence_threshold,
+                         double iou_threshold)
+        : BoxesLabelsScoresConverter(std::move(initializer), confidence_threshold, iou_threshold) {
         const GstStructure *s = getModelProcOutputInfo().get();
         gboolean do_cls_sftm = FALSE;
         if (gst_structure_has_field(s, "do_cls_softmax")) {