基于onnx进行yolov5推理

img

如图,我的onnx模型输出是YOLOV5中的3个head,网上关于onnx推理YOLOV5的大多是融合head后的。
对于输出为3个head的,该如何进行onnx推理?

Yolo v5 pytorch模型转onnx用c++进行推理:https://blog.csdn.net/qq_33671888/article/details/114284159

3个的话分开推理就行啊,最早版本的yolov5就是分开的,你可以看下我的yolov7目前就是分开推理的,合并推理会报错。

源代码:

img