整个项目配置过程和教程讲解过程的录制视频,可搜索b站或西瓜视频up:杭漂一族小张。
有需demo的可联系扣扣2014453702
这个专栏更新各种AI,以及各种有趣的教程,有兴趣的小伙伴可以订阅一下。
首先是图片的搜集,方法无非两种
一是开源的数据集,直接拿来用,另一种就是自己搜集图片制作数据集。
这里我们可以通过爬虫,或是自己通过其他途径获取数据,我这里以水果识别为例子
爬虫搜集5种水果各若干,放在data/各个水果文件夹下。
比如草莓这个图片是这样的,我们只要准备好图片,放到对应文件夹下即可。
然后需要我们对图片文件夹就行制作数据集,共训练模型时,方便读取,这里以生成txt文本为例子。