重现步骤
在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