yolo大作战onnx模型转换kmodel模型时报错

Viewed 422
yolo k230

重现步骤

期待结果和实际结果

软硬件版本信息

nncase 2.9.0

错误日志

尝试解决过程

补充材料

shanwoo5
2 Answers

现在用onnx1.17.0在转换成onnx模型这一步就报错了不过把onnx的版本降到1.16.1就能顺利转换成kmodel模型了

shanwoo5

按理说16应该也没问题，你这个是转onnx出的问题，onnx内部都是NAN

Wy001•

能否从头搭一下环境，按照yolo官方的要求搭建环境，yolo官方提供了requirements.txt

Wy001•

按你说的搭了，还是得用onnx1.16.1才行

shanwoo•

您好，请给一下onnx模型。

@shanwoo你这个模型是否经过训练，是不是训练参数有问题，为啥使用netron打开的onnx模型的权重都是NAN，转换脚本应该没问题，可能是你模型有问题。你换个模型试试。
image.png

================================================
脚本转换的正确onnx模型信息：
image.png

Wy0012026

python classify/train.py --model yolov5n-cls.pt --data datasets/fruits_cls --epochs 100 --batch-size 8 --imgsz 224 --workers 4 --device 0
用这条语句训练的

shanwoo•

我用netron打开best.pt文件，里面running_mean和running_var的值是NAN，是这里的问题吗？

shanwoo•

训练过程正常吗

Wy001•