Atlas系列硬件: Atlas 200l DK A2 开发者套件; Atlas 200l DK 开发者套件（型号：3000）; 加速模块、加速卡、智能小站、服务器、集群

异构计算架构: CANN社区版; CANN商用版

AI框架: MindSpore; TensorFlow框架适配（社区版）; TensorFlow框架适配（商用版）; PyTorch框架适配

分布式训练加速套件: MindSpeed

昇腾推理引擎: MindIE

应用使能MindX: MindX DL; MindX Edge; MindX SDK

全流程开发工具: MindStudio

评分

提单

论坛

小AI

YoloNms

输入

x1：

是否必填：必填
数据类型：float16
参数解释：Yolov5FourInputDecodeBox的输出结果
规格限制：NCHW

属性

shape：
- 是否必填：必填
- 数据类型：int
- 参数解释：输入Yolov5FourInputDecodeBox算子的4个feature_map的h*w的和

thresh：
- 是否必填：必填
- 数据类型：float16
- 参数解释：nms的阈值
- 规格限制：0.0-1.0之间

num_anchor：
- 是否必填：必填
- 数据类型：int
- 参数解释：anchor数
- 规格限制：大于0

num_class：
- 是否必填：必填
- 数据类型：int
- 参数解释：类别数
- 规格限制：1到80类

total_output_proposal_num：
- 是否必填：必填
- 数据类型：int
- 参数解释：每个检测的类别要输出的框的个数
- 规格限制：5到50之间

输出

y：

是否必填：必填
数据类型：float16
参数解释：ssmh网络后处理进行nms之后的输出结果，[total_output_proposal_num*class_num, 8]
规格限制：NCHW，输出的框是按类别排布的，[total_output_proposal_num*class_num, :4]是坐标，[total_output_proposal_num*class_num, 4]是得分，[total_output_proposal_num*class_num, 5：]是保留位

父主题： 支持Caffe算子清单