NmsSort

Filters repeated objects based on the confidence score.

void NmsSort(std::vector<DetectBox>& detBoxes, float iouThresh, IOUMethod method = UNION);

void NmsSort(std::vector<ObjectInfo>& detBoxes, float iouThresh, IOUMethod method = UNION);

Parameter	Input/Output	Description
detBoxes	Output	DetectBox (central point)/ObjectInfo (corner point) to be filtered. The filtered object information is output.
iouThresh	Input	Filtering threshold.
method	Input	IoU is calculated as follows: MAX: The overlapped area is divided by the larger one of the two areas. MIN: The overlapped area is divided by the smaller one of the two areas. UNION: The overlapped area is divided by the union of the two areas. DIOU: The value equals the value of UNION minus the IoU of the distances.

Parameter

Input/Output

Description

detBoxes

Output

DetectBox (central point)/ObjectInfo (corner point) to be filtered. The filtered object information is output.

iouThresh

Input

Filtering threshold.

method

Input

IoU is calculated as follows:

MAX: The overlapped area is divided by the larger one of the two areas.
MIN: The overlapped area is divided by the smaller one of the two areas.
UNION: The overlapped area is divided by the union of the two areas.
DIOU: The value equals the value of UNION minus the IoU of the distances.

Parent topic: Nms