BatchMultiClassNMS

功能

为输入boxes和输入score计算nms。

输入

两个必选输入，两个可选输入：

boxes：必选，一个tensor，数据类型：float16、float32。

scores：必选，一个tensor，数据类型：float16、float32。

clip_window：可选，一个tensor，数据类型：float16、float32。

num_valid_boxes：可选，一个tensor，数据类型：int32。

输出

四个输出：

nmsed_boxes：一个tensor，数据类型：float16、float32。

nmsed_scores：一个tensor，数据类型：float16、float32。

nmsed_classes：一个tensor，数据类型：float16、float32。

nmsed_num：一个tensor，数据类型：int32。

属性

score_threshold：float型。

iou_threshold：float型。

max_size_per_class：int型。

max_total_size：int型。

change_coordinate_frame：bool型。

transpose_box：bool型。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16