k230-YOLO26s转换kmodel后测试Cosine 相似度一直只有0.5几

Viewed 6

问题描述


用自标注素材训练完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

1 Answers

Yolo26的架构把nms加入到模型里面了,导致输出中存在很大的坐标,不是纯浮点输出。所以微小的差异会造成相似度很差,直接测试一下推理结果,如果坐标框差异不大就没问题。