nmsyolov5

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

Supports both YoloV5 (masks) and YoloV7 (bounding boxes) models. It is achieved by utilizing the same memory structure (in case of YoloV7 mask information will be ingnored). Performs the following steps: #. Filtering input BBoxes by confidence filter. #. Applying TopK algorithm on filtering result. #. Running NMS algorithm on the TopK result. #. Applying the TopK algorithm on the result of NMS.

Usage

Example of usage:

gst-launch-1.0 --gst-plugin-path=/path/to/your/plugin/lib simaaisrc location=/data/yolo_postproc.bin node-name="yolopostproc" mem-target=1 ! nmsyolov5 config=/data/config_nms.json ! fakesink

Configuration

This plugin is not configurable at this time.