2301_79743797的博客

转载 yolov5在onnx转rknn时推理结果框多且乱的解决方法

注意：网上一顿查找发现是在将pt文件转化为onnx时对models/yolo.py的修改有问题。网上大部分的修改都是下面这种，这种可能有问题。题主打算将yolov5部署在香橙派中，将onnx转rknn模型时出现了框多且乱的情况，但是最终解决了。export.py文件的run函数。首先要修改yolo.py的代码。然后修改export.py文件。然后转为onnx模型即可。