自己训练的yolov5模型 量化后 推理全是密密麻麻的检测框


求助

两种可能:1.输出头没改 2.后处理维度不匹配。

论坛里面说了,用model_zoo里面的流程比较好。我用了x86文件夹中,mapper.py转换模型,nv12输入,输出改成NWHC,再使用cpp推理,就正常了。