yolo5模型转换kmodel报错

Viewed 51

重现步骤
按照文档步骤
yolo5模型已经训练完成,转换pt到onnx模型的时候提示异常
python export.py --weight runs/train-cls/exp/weights/best.pt --imgsz 224 --batch 1 --include onnx

ONNX: export failure 2.5s: DLL load failed while importing onnx_cpp2py_export: (DLL)

期待结果和实际结果
pt转成onnx格式

1 Answers

可能是onnx版本和系统不兼容,降一下onnx版本

python3.8.6 用onnx什么版本?nncase nncase-kpu都是2.9.0
pip install nncase==2.9.0
pip install nncase-kpu==2.9.0

这和nncase无关,把onnx版本降一下试试