【原创】【踩坑日记】onnx转ncnn模型出bug

python导出的onnx格式的yolo模型为yolov11.onnx文件。输出代码如下:

from ultralytics import YOLO
model = YOLO("yolo11n.pt")
success = model.export(format="onnx")

ncnn需要两个模型文件才能推理,.param和.bin文件。
官方建议使用onnx2ncnn.exe导出,这个导出的onnx格式文件(yolov11.onnx)直接转ncnn会出问题(onnx2ncnn.exe)。
onnx2ncnn.exe yolov11.onnx yolov11.param yolov11.bin

实际上ultralytics 官方有自己的ncnn模型输出方法:https://docs.ultralytics.com/integrations/ncnn/#why-should-i-use-ncnn-for-my-mobile-ai-applications

from ultralytics import YOLO

# Load the YOLO11 model
model = YOLO("yolo11n.pt")

# Export to NCNN format
model.export(format="ncnn")  # creates '/yolo11n_ncnn_model'

这样直接会导出bin和param文件。

此外ncnn官方也建议抛弃onnx2ncnn.exe:
https://github.com/Tencent/ncnn/issues/5508

posted @   范哥范小飞  阅读(222)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 提示词工程——AI应用必不可少的技术
· 地球OL攻略 —— 某应届生求职总结
· 字符编码:从基础到乱码解决
· SpringCloud带你走进微服务的世界
主题色彩
点击右上角即可分享
微信分享提示