nms

Non-maximum suppression (NMS) is a post-processing technique that is commonly used in object detection tasks. This plugin is used to eliminate duplicate detections and to select the most relevant bounding boxes that correspond to the detected objects.

Features

The current NMS algorithm library provides a subset of NMS functions that can be used:

  • nms_simd (SIMD optimized algorithm)

  • yolo_nms (YOLO specific implementation)

  • nms (non-optimized version of algorithm + TopK algorithm included)

Usage

The current implementation of the nms plugin is specific to the EfficientDet model. This plugin uses the simaai::pipeline_utils::invert_affine function to upscale bounding boxes. You may need to remove this function in your specific pipeline.

1 gst-launch-1.0 --gst-plugin-path=/path/to/your/plugin/lib simaaisrc location=/data/nms_in.bin node-name="transform-anchors" mem-target=0 ! nms config=/data/config_nms.json ! fakesink

Configuration

This plugin is not configurable at this time.