问题描述
用自标注素材训练完yolo26s后导出onnx在转换Kmodel,素材近万,抽900多作为转换时校准素材。过程都是顺利的,不过使用 nncase 模拟器验证转换效果时得到结果一直只有0.5几,尝试使用不同参数转换kmodel结果都差不多。使用小批次素材对.pt和.onnx模型文件进行测试结果几乎一致,用test_yolo26工具detect目录下的test_det_kmodel测试kmodel文件结果差一些,置信度低不少。有大佬知道是什么问题吗。
硬件板卡
k230mini
其他信息
环境部署和操作步骤都是参考:https://www.kendryte.com/k230_canmv/zh/main/example/ai/yolo_battle.html#yolo26
测试环境:windows10
python:3.9
nncase和nncase-kpu版本(离线安装):2.11.0
导出Onnx命令:yolo export model=best.pt format=onnx imgsz=320 nms=False simplify=True
转换kmodel命令:py -3.9 to_kmodel.py --target k230 --model ../../26s/best.onnx --dataset ../../26s/datasets_images_test --input_width 320 --input_height 320 --ptq_option 0