YOLOV8 如何更改成为合适的输出头


本人纯小白,求大神指教!!

在将yolov8.n模型上板时一直报错,后来知道是输出头的顺序不对,反量化系数也要一样,要更改append的顺序才可以但我不会,想请教一下

下面提供我的onnx模型可视图和官方的onnx模型可视图,附录提供我的head.py文件,想知道具体怎么解决有没有文献可以参考

我的onnx可视图

官方的:


head.py

提供head是没用的,参考README去修改cv2,cv3,cv4的顺序,然后hb_mapper checker 的顺序去看,bin输出顺序一定和onnx一样

超哥,这个README是model_zoo里yolov8下的README吗?我怎么找不到

model_zoo里面一定有的

现在维护的比较好的是YOLO11,两个是一样的