原作者链接:基于卷积神经网络的中药识别(pytorch框架)【python源码+UI界面+前端界面+功能源码详解】_识别中药python-CSDN博客
//gitcode,gitee,飞桨,csdn,bilibili。几个有用网站,直接搜索即可,平常做项目,找数据集,找代码都可以去上面搜搜看,遇到一些问题,可以去csdn找合适答案。(当然,好的东西肯定是需要收费的)
//本文只补充调试该项目可能遇到的问题。
1.工具
pycharm+annconda/python(python==3.8)
2.问题
(1)
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True
(2)数据集路径中有中文
本人的数据集都是英文命名,如果遇到这类问题,要么把文件名以英文命名,要么csdn搜索一下,修改部分代码即可。
3.总结
该作者的数据集只有5种中药(还有一个是24种),但是这里可以自己去飞桨或其他平台获取你需要的数据集,修改相关模型的对应路径即可。并且该项目不只是可以中药材识别,也可以垃圾识别,鸟类识别,只需要修改数据集就行,但是不同的模型对数据集的训练效果可能不一样。
//下面的链接是如何划分训练集、测试集、验证集。
如何划分训练集、测试集、验证集_训练集、验证集和测试集的划分-CSDN博客
4.未完待续