Initialization

Parameters

The NMSAgent is initialized with two arguments:

NMSAgent(iou_threshold, score_threshold)
iou_threshold
Float
required

This 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
required

This 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)