BatchMultiClassNMS

功能

为输入boxes和输入score计算nms

边界

【输入】

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

boxes：一个tensor，数据类型：float16

scores：一个tensor，数据类型：float16

clip_window：一个tensor，数据类型：float16

num_valid_boxes：一个tensor，数据类型：int32

【属性】

六个属性：

score_threshold：float型

iou_threshold：float型

max_size_per_class：int型

max_total_size：int型

change_coordinate_frame：bool型

transpose_box：bool型

【输出】

四个输出

nmsed_boxes：一个tensor，数据类型：float16

nmsed_scores：一个tensor，数据类型：float16

nmsed_classes：一个tensor，数据类型：float16

nmsed_num：一个tensor，数据类型：float16

支持的ONNX版本

自定义算子，无对应onnx版本