CANN/ops-cv非极大值抑制V3算子

NonMaxSuppressionV3【免费下载链接】ops-cv本项目是CANN提供的图像处理、目标检测相关的算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-cv产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品×Atlas 推理系列产品√Atlas 训练系列产品√功能说明算子功能按照分数递减顺序采用贪心策略选择候选框bounding boxes子集。参数说明参数名输入/输出/属性描述数据类型数据格式boxes输入二维 Tensor其 shape 为 (num_boxes, 4)。输入格式为 (x1, y1, x2, y2)并且要求 x1 x2 且 y1 y2。FLOAT16、FLOATNDscores输入一维 Tensor其 shape 为 (num_boxes)。表示与每个候选框对应的分数与 boxes 的每一行一一对应。FLOAT16、FLOATNDmax_output_size输入标量整数 Tensor表示非极大值抑制Non-Maximum Suppression最多选择的候选框数量。INT32NDiou_threshold输入标量浮点 Tensor表示用于判断候选框是否在 IoU交并比上重叠过多的阈值。FLOAT16、FLOATNDscore_threshold输入标量浮点 Tensor表示用于根据分数移除候选框的阈值。FLOAT16、FLOATNDoffset可选属性• 可选整数。• 默认值为 0。INT-selected_indices输出一维整数 Tensorshape 为 (M)表示从输入 boxes Tensor 中选中的索引其中 M max_output_size。INT32ND约束说明输入的 boxes 和 scores 必须是 float 类型。调用说明调用方式调用样例说明图模式调用test_geir_non_max_suppression_v3通过算子IR构图方式调用NonMaxSuppressionV3算子。【免费下载链接】ops-cv本项目是CANN提供的图像处理、目标检测相关的算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-cv创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考