search

airockchip / rknn_model_zoo

Apache License 2.0
1.04k stars 190 forks source link

请问Yolov8的模型怎么转成提供的.onnx文件的？ #59

Open HY59927 opened 10 months ago

HY59927 commented 10 months ago

我发现该工程Yolov8文件夹提供的n，s，m模型的onnx文件有三层输出，而官网提供的脚本转出来的是一个输出层为(1, 84, 8400)的onnx结构，怎么才能转成你们提供的onnx结构呢？ 官网提供的脚本转出来的onnx虽然能成功转换成rknn，但是推理会报错。万分感谢。

HY59927 commented 10 months ago

找到答案了，在这个视频里，https://www.youtube.com/watch?v=pN8mKZ5wpdQ，sencond step里有说

15116513319 commented 10 months ago

请问有中文的解决方案吗？

kaylorchen commented 8 months ago

我做了一个易用的教程，pt转onnx，然后onnx转rknn， 参考我的文章https://blog.kaylordut.com/2024/02/09/rk3588's-yolov8-model-conversion-from-pt-to-rknn/

zxt-triumph commented 7 months ago

请问怎么导出三个输出的onnx啊？我用yolov8官网的方式只有一个输出

pubuzai commented 2 months ago

找到答案了，在这个视频里，https://www.youtube.com/watch?v=pN8mKZ5wpdQ，sencond step里有说

感谢，按照视频的方法，成功转出和官方提供onnx一致的检测头的onnx，视频13分30秒左右有讲yolov8如何转onnx的示例，照着视频步骤能转成功

Eliza-and-black commented 2 months ago

我做了一个易用的教程，pt转onnx，然后onnx转rknn， 参考我的文章https://blog.kaylordut.com/2024/02/09/rk3588's-yolov8-model-conversion-from-pt-to-rknn/

Very useful part in your web about the commit id. Tks.