【原创】【踩坑日记】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 @ 2024-12-24 21:53  范哥范小飞  阅读(1001)  评论(0)    收藏  举报