阿白_nano · 2023年11月07日 · 福建

【聆思CSK6 LNN工具体验】自定义数据集的分类模型训练及产品使用踩坑

基于自定义数据集的咖啡豆颜色分类

生成和cifar-100一样格式的数据集

数据集有4类,分别为Dark,Green,Light,Medium
image.png
image.png
image.png
image.png
image.png

06BAD69B47BB8E94EC522BE78AF9F2B6.png
2BEBA7D9205B88C9A56A975DEEA8BB4E.png

上面这段代码可以生成train和val2个文件
image.png

模型训练

根据原作者的代码写一个Dataset类
image.png

将模型部署到板子上

这个根据官方文档苹果识别的例子,修改一些参数即可
image.png
注意修改class,和模型大小。
以下是串口输出结果:
image.png

碰到的问题

问题一

第一个就是环境问题,我的云环境是ubuntu18.04,cuda10.2,显卡是v100。本机是ubuntu20.04,cuda11. 30系列的只测试了3090不支持

问题二

image.png
解决方法
image.png

问题三

version `GLIBC_2.34‘ not found简单有效解决方法
https://blog.csdn.net/huazhan...
如何解决version `GLIBCXX_3.4.29‘ not found的问题
https://blog.csdn.net/weixin_...

后续的改进

把图像分类改成实时的图像分类,有时间再搞。

推荐阅读
关注数
0
文章数
1
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息