YOLO模型后处理
收藏回复举报
YOLO模型后处理
t('forum.solved') 已解决
发表于2025-03-31 14:02:03
0 查看
使用yolov5的模型推理得到输出张量(1,25200,6),准备使用后处理接口YoloV3PostProcess,其配置文件如下所示:
cke_3084.png

运行得到如下报错:
cke_5093.png

有几个不明白的地方:
1.官方提供的检测后处理接口对模型推理输出张量的形状要求是怎样的,有没有示例
2.如果自己写后处理的模块,如非极大值抑制等有硬件加速的接口嘛

我要发帖子