开发者
资源
[object Object][object Object][object Object]undefined
[object Object]

执行批量多类别的非极大抑制(NMS)计算。

[object Object]
[object Object]
[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]

[object Object]

表示候选框和标签(bboxes, labels),shape为(bs, k, 5)和(bs, k)的张量。标签以0为基础。

[object Object]

在动态shape条件下,最多支持20个类别(nmsed_classes)和10000个框(nmsed_boxes)。

[object Object]
[object Object]