Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
Yolo算子支持ops-cv
Yolo算子是YOLO v2/v3目标检测网络的后处理层,对检测特征图进行数据重组和激活处理,将卷积输出转换为坐标、目标置信度和类别概率三个输出。
算子特点:
4种计算模式(YOLO_MODE_1/2/3/4),由yolo_version+softmax+background决定 sigmoid/softmax/move三种激活方式 w/h坐标位置交换 float16/float32精度,中间计算提升到float32 实现方式:AscendC SIMT,多核多线程并行(GetBlockIdx/GetBlockNum + 512线程/核 + Grid-Stride)
华为海思昇腾
Yolo算子支持Ascend950
实现方式:AscendC SIMT,多核多线程并行(GetBlockIdx/GetBlockNum + 512线程/核 + Grid-Stride)
/assign
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
Backgroud(背景信息)
Yolo算子支持ops-cv
Yolo算子是YOLO v2/v3目标检测网络的后处理层,对检测特征图进行数据重组和激活处理,将卷积输出转换为坐标、目标置信度和类别概率三个输出。
算子特点:
4种计算模式(YOLO_MODE_1/2/3/4),由yolo_version+softmax+background决定
sigmoid/softmax/move三种激活方式
w/h坐标位置交换
float16/float32精度,中间计算提升到float32
实现方式:AscendC SIMT,多核多线程并行(GetBlockIdx/GetBlockNum + 512线程/核 + Grid-Stride)
Origin(信息来源)
华为海思昇腾
Benefit / Necessity (价值/作用)
Yolo算子支持Ascend950
Design(设计方案)
实现方式:AscendC SIMT,多核多线程并行(GetBlockIdx/GetBlockNum + 512线程/核 + Grid-Stride)