Detection Agents
NMS Agent
Detection Agents
NMS Agent
The NMSAgent is used to perform Non-Maximum Suppression (NMS) on a set of detections, specifically for bounding boxes.
Initialization
Parameters
The NMSAgent is initialized with two arguments:
NMSAgent(iou_threshold, score_threshold)
iou_threshold
Float
requiredThis is the Intersection Over Union (IOU) threshold used to determine whether two bounding boxes overlap more than the specified threshold. If they do, the one with the lower confidence score is suppressed.
score_threshold
Float
requiredThis is the minimum confidence score that a detection must have to be considered valid. Detections with scores below this threshold are ignored.
Example
example.py
NMSAgent(iou_threshold=0.5, score_threshold=0.6)