NonMaxSuppression

功能

从众多重叠的候选边界框中筛选出最优的边界框,过滤掉与先前选定的框有较高重叠的“交集-并集”(IOU)框。移除得分小于score_threshold的边界框。边界框格式由属性center_point_box表示。

该算法与坐标系原点无关,对坐标系的正交变换和平移操作保持不变。输出selected_indices为整数集,可通过Gather或GatherND操作获取对应边框坐标。

输入

属性

center_point_box:数据类型为 int,默认值为0,决定了边界框格式。

输出

selected_indices:输出Tensor,数据类型:int64。

约束与限制

支持的ONNX版本

Opset v11/v12/v13/v14/v15/v16/v17/v18。