如何利用CANN算子进行图构建,生成后处理算子?
收藏回复举报
如何利用CANN算子进行图构建,生成后处理算子?
t('forum.solved') 已解决
发表于2024-03-06 22:28:19
0 查看

参考案例inference/modelInference/sampleYOLOV7NMSONNX · Ascend/samples - 码云 - 开源中国 (gitee.com)的写法(利用CANN算子进行图构建,生成后处理算子),想为YOLOX生成后处理NMS算子,查CANN算子清单有疑似支持的算子BatchMultiClassNonMaxSuppression和YoloxBoundingBoxDecode(),想问下这个思路的可行性?

cke_2608.pngcke_12818.png

我要发帖子