如图,我的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目前就是分开推理的,合并推理会报错。
源代码: