大家好，马上要到中秋节了，今天给大家分享一个非常适合新手的Ai小项目—月饼🥮检测器，跟着我的博文，我会带你一步一步的从零开始实现属于自己的月饼🥮检测器
第一步：准备一些月饼图片
我这里让朋友帮忙爬取了600张，清洗筛选后最终得到了304张
第二步：标注数据
由于Labelimg和Labelme安装起来比较麻烦，所以这次我选择使用make sense：https://www.makesense.ai
这是一款在线标注数据集的工具，用起来非常的方便
进入网页后点击Get Started开始使用
首先点击Drop images然后Ctrl+A选中整个数据集里面的图片
随后添加标签信息，因为我这里只检测月饼一类，所以只添加一个标签 Moon Cake
随后就进入了漫长的标注环节，这里大家一定要认真标注，不然对最终模型的影响还是很大的
大约3个小时以后就标注完毕了。。。。
我们点击Action->Export Annotation 导出yolo格式的标签文件
导出之后的标签文件就是这个样子的
第三步：划分数据集
导出后按一定比例划分训练集和验证集，我这里训练集选择了273张图片，验证集选择了31张图片
按照以下格式放置：
数据集准备好之后就可以着手训练模型啦
第四步：训练模型
我们这次选择yolov5模型作为基线训练
首先去yolov5的官方项目里下载代码：https://github.com/ultralytics/yolov5
也可以使用我个人基于yolov5官方项目改进的项目Yolov5_Magic：https://github.com/WangQvQ/Yolov5_Magic
我这里选择我个人改进的yolov5项目
首先在data的文件夹里新建一个MoonCake.yaml，我们要在这个MoonCake.yaml文件里放上我们数据集的路径和类别信息
MoonCake.yaml我们按照如下的格式书写，因为没有划分测试集，所以这里我们可以不写test的路径
新手我推荐直接写绝对路径，这样可以减少一些错误
接下来我们在项目里找到train.py ，在--data这里放上我们刚刚写好的MoonCake.yaml的路径
因为数据集比较少，所以我这里选择了轻量化的主干网络shufflenetv2，如果你选择的是我的代码，那么你会在models里发现一个写好的配置文件yolov5_ShuffleNetv2.yaml，如果你不选择我的代码，那么你使用默认的yolov5s就可以了；
所以在这里放上yolov5_ShuffleNetv2的路径，当然用默认的也是可以的
随后我们就可以开始训练模型啦
各个参数更加详细的解释可以看我的其它博文：手把手带你调参Yolo v5 (v6.1)（二）
参数简要说明：
–weights：初始化的权重文件的路径地址
–cfg：模型yaml文件的路径地址
–data：数据yaml文件的路径地址
–hyp：超参数文件路径地址
–epochs：训练轮次
–batch-size：喂入批次文件的多少
–img-size：输入图片尺寸
–rect:是否采用矩形训练，默认False
–resume:接着打断训练上次的结果接着训练
–nosave:不保存模型，默认False
–notest:不进行test，默认False
–noautoanchor:不自动调整anchor，默认False
–evolve:是否进行超参数进化，默认False
–bucket:谷歌云盘bucket，一般不会用到
–cache-images:是否提前缓存图片到内存，以加快训练速度，默认False
–image-weights：使用加权图像选择进行训练
–device:训练的设备，cpu；0(表示一个gpu设备cuda:0)；0,1,2,3(多个gpu设备)
–multi-scale:是否进行多尺度训练，默认False
–single-cls:数据集是否只有一个类别，默认False
–adam:是否使用adam优化器
–sync-bn:是否使用跨卡同步BN,在DDP模式使用
–local_rank：DDP参数，请勿修改
–workers：最大工作核心数
–project:训练模型的保存位置
–name：模型保存的目录名称
–exist-ok：模型目录是否存在，不存在就创建
我们可以在终端里键入：python train.py --epoch 100
当出现这样的画面的时候就说明开始训练啦，我们只要耐心的等待结果就可以了
我的设备是3060显卡，100轮12分钟就训练好了，最终准确率达到了86.5%，召回率达到了78.7%，效果还是很不错的，最终的模型数据就保存在红框中的路径里，记住这里，一会儿会用得到
接下来我们就可以找一些图片测试一下我们的模型实际的检测效果了
