基于自定义数据集的咖啡豆颜色分类
生成和cifar-100一样格式的数据集
数据集有4类,分别为Dark,Green,Light,Medium
上面这段代码可以生成train和val2个文件
模型训练
根据原作者的代码写一个Dataset类
将模型部署到板子上
这个根据官方文档苹果识别的例子,修改一些参数即可
注意修改class,和模型大小。
以下是串口输出结果:
碰到的问题
问题一
第一个就是环境问题,我的云环境是ubuntu18.04,cuda10.2,显卡是v100。本机是ubuntu20.04,cuda11. 30系列的只测试了3090不支持
问题二
解决方法
问题三
version `GLIBC_2.34‘ not found简单有效解决方法
https://blog.csdn.net/huazhan...
如何解决version `GLIBCXX_3.4.29‘ not found的问题
https://blog.csdn.net/weixin_...
后续的改进
把图像分类改成实时的图像分类,有时间再搞。