yolov8训练模型询问

Viewed 72

重现步骤
在vscode上用yolov8训练自己的数据集,迭代400次,训练完后验证集识别置信度在0.9左右,但将pt格式的模型转化为onnx格式再转为kmodel格式后导入摄像头中,置信度始终在0.503,且近距离才能识别到,图片分辨率为224x224

期待结果和实际结果
远距离(0.5米左右也能识别到)且置信度>=0.85
只有2分米以内才能识别到且置信度始终保持0.503不变

软硬件版本信息
k230 yolov8

错误日志

尝试解决过程

补充材料

3 Answers

我也想问这个转换的损失(识别准确度下降)怎么降低?

可以在转模型的时候把ptq_option设置为1或者2试试。kmodel模型转换的过程中是有量化损失的。另外你可以用同一张图片测一下,看看图片推理的结果对不对,因为视频可能和你的数据有差异,但是同一张图用onnx和kmodel的测试应该改差距不大。

修改了ptq后好像没有变化,我的数字都能正常识别,但最大置信度一直是0.503