aclnnNonMaxSuppression
支持的产品型号
- Atlas 推理系列产品。
接口原型
每个算子分为，必须先调用“aclnnNonMaxSuppressionGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnNonMaxSuppression”接口执行计算。
aclnnStatus aclnnNonMaxSuppressionGetWorkspaceSize(const aclTensor *boxes, const aclTensor *scores, aclIntArray *maxOutputBoxesPerClass, aclFloatArray *iouThreshold, aclFloatArray *scoreThreshold, int32_t centerPointBox, aclTensor *selectedIndices, uint64_t *workspaceSize, aclOpExecutor **executor)aclnnStatus aclnnNonMaxSuppression(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
功能描述
算子功能：NonMaxSuppression, 筛选出 与之前被选中的有较高交集并集（IOU） 重叠的框, 删除分数小于 scoreThreshold 的边界框。
aclnnNonMaxSuppressionGetWorkspaceSize
参数说明：
boxes(aclTensor*, 必选输入)：Device侧的aclTensor，数据类型支持FLOAT、FLOAT16。支持，支持。shape为[num_batches, spatial_dimension, 4]。
scores(aclTensor*, 必选输入): Device侧的aclTensor，数据类型支持FLOAT、FLOAT16。支持，支持。shape为[num_batches, num_classes, spatial_dimension]。
maxOutputBoxesPerClass(aclIntArray*, 可选输入): Host侧的aclIntArray，数据类型支持INT32。支持，支持。标量，默认值为0，表示无输出。表示每个类别每batch要选择的最大框数。
iouThreshold(aclFloatArray*, 可选输入): Host侧的aclFloatArray，数据类型支持FLOAT。支持，支持。标量，默认值为0，取值范围[0, 1]。表示判断框相对于IOU是否重叠过多的阈值。
scoreThreshold(aclFloatArray*, 可选输入): Host侧的aclFloatArray，数据类型支持FLOAT。支持，支持。标量。表示根据得分决定何时移除框的阈值。
centerPointBox(int, 计算输入): Host侧的整数，数据类型支持INT32。取值范围0或1。默认值为0。作为属性 它决定了边界框格式。等于0时，主要用于TF模型, 数据以(y1, x1, y2, x2)形式提供，其中(y1, x1) (y2, x2)是对角线框角坐标，其中x1<x2, y1<y2 需要用户自行保证; 等于1时, 主要用于pytorch模型。框数据以(x_center, y_center, width, height)形式提供。
selectedIndices(aclTensor*, 计算输出)：Device侧的aclTensor，数据类型支持INT64。支持，支持。输出的是一组整数，索引到表示所选框的边界框的输入集合中
workspaceSize(uint64_t*, 出参)：返回用户需要在Device侧申请的workspace大小。
executor(aclOpExecutor**, 出参)：返回op执行器，包含了算子计算流程。
返回码：
aclnnNonMaxSuppression
参数说明：
workspace(void*, 入参)：在Device侧申请的workspace内存地址。
workspaceSize(uint64_t, 入参)：在Device侧申请的workspace大小，由第一段接口aclnnNonMaxSuppressionGetWorkspaceSize获取。
executor(aclOpExecutor*, 入参)：op执行器，包含了算子计算流程。
stream(aclrtStream, 入参)：指定执行任务的 AscendCL Stream流。
返回码：
约束与限制
无。