K230云平台训练的分类模型识别结果全为1.0

Viewed 187

问题描述
我自己为鼠标和笔拍了照片,在云平台上训练了分类模型,但是我将训练得到的压缩包解压后,按照教程放到开发板上跑,识别结果全是鼠标,并且为1.0。

软硬件条件
我的硬件为01Studio的K230开发板,固件为01Studio提供的01Studio_CanMV_K230_MicroPython_v1.0.img(感觉应该是对应嘉楠官网的CanMV-K230_01Studio_micropython_v1.0_sdcard_v1.6_nncase_v2.8.3.img.gz)

模型训练的关键动作及结果
模型训练时,数据集里鼠标和笔我各添加了160张,训练时选择了K230,nncase为2.8.3(这个版本应该是正确的,不然后面我应该也不能运行模型),迭代次数等其他参数都是默认
训练结果如下,训练结果应该不错,到后面loss很低很低

运行模型和结果
然后就是按照教程,复制文件到SD卡,并运行video的py代码(没有做任何修改,是直接运行的),但是很奇怪我无论拍什么他都说识别到了目标,并且为1.0


并且我也尝试过image的那个py文件,结果一样
我也尝试过修改json文件,把confidence_threshold改到0.9,依旧是老样子。

想请教一下大家有没有遇到和我类似问题的?
还有JSON文件中mean、std这两个参数是什么意思?

我训练结果的压缩包因为论坛没有找到上传附件的地方,就没法分享出来,video的代码应该是官方的一个通用模板,感觉应该是不会有问题的,感觉应该还是我训练模型和使用模型那边有问题

2 Answers

我还尝试了官网提供的分类数据集,把里面的菠菜和胡萝卜拿去训练了,结果依旧

尝试下载官网01studio开发板的1.0版本(和我之前刷的应该是同一个,但是拿官网的再刷一次,排除一下镜像的问题),现象依旧

已修改代码,您可以重新提交任务训练测试

非常感谢,我马上去试试

非常抱歉,这是我们提供代码的问题,后续我们会尽快修改。谢谢