执行批量多类别的非极大抑制(NMS)计算。
[object Object]
- multi_bboxes (
[object Object]): 必选参数。候选框(bbox)张量,shape为(bs, n, class, 4)或(bs, n, 4)。 - multi_scores (
[object Object]): 必选参数。每个候选框的类别得分,shape为(bs, n, class+1),其中最后一列包含background class分数,可忽略。 - score_thr (
[object Object]): 可选参数,默认值为0.05。候选框分数阈值,分数低于它的候选框将不被考虑。 - nms_thr (
[object Object]): 可选参数,默认值为0.45。NMS IoU阈值。 - max_num (
[object Object]): 可选参数,默认值为50。如果NMS后的bbox数超过max_num值,则只保留max_num个bbox;如果NMS后的bbox数小于max_num值,则输出将零填充到max_num值。 - score_factors (
[object Object]): 可选参数,默认值为None。NMS应用前用来乘分数的因子。
[object Object]
表示候选框和标签(bboxes, labels),shape为(bs, k, 5)和(bs, k)的张量。标签以0为基础。
在动态shape条件下,最多支持20个类别(nmsed_classes)和10000个框(nmsed_boxes)。
[object Object]