【原创】【踩坑日记】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
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 提示词工程——AI应用必不可少的技术
· 地球OL攻略 —— 某应届生求职总结
· 字符编码:从基础到乱码解决
· SpringCloud带你走进微服务的世界