在部署kmodel模型时出现该报错ValueError:array is too big

Viewed 78

重现步骤

在Windows使用本地部署的yolo5跑出模型转换后得到kmodel模型,然后部署在K230上的时候IDE出现报错ValueError:array is too big!

期待结果和实际结果

软硬件版本信息

k230固件1.3.5 使用yolov5n

错误日志

尝试解决过程

补充材料

代码截图

yolov5

pt转onnx指令
python export.py --weight runs/train/exp12/weights/best.pt --imgsz 224 --batch 1 --include onnx

onnx转kmodel指令
python to_kmodel.py --target k230 --model ../../runs/train/exp12/weights/best.onnx --dataset ../test --input_width 224 --input_height 224 --ptq_option 0

1 Answers

有没有一种可能,你拷贝的是分类任务的代码

我改成了图像识别的代码还是报相同的错误

你好 请问你的问题是否解决了呢