一、什么是yolov4?
YOLOv4是一种目标检测算法,是YOLO (You Only Look Once) 系列的最新版本,由Alexey Bochkovskiy、Chien-Yao Wang和Hong-Yuan Mark Liao共同提出。相比于之前的版本,YOLOv4在速度和精度方面都有了显著的提升。
二、数据集准备
我是在网上搜索的三种水果照片创建了自己的数据集,不想自己动手的话也可以在站内使用别人的,不过我没有找到免费的,就自己收集了。但是耗时比较长,工作繁杂,我一共找了160张。
三、训练模型
1、将数据集放到指定文件位置
把JPEGImages文件夹原来的数据集删除,将自己的数据集放到JPEGImages文件内。
2、设置标签
运行老师给的labelImg程序,打开数据集图片,自行框选目标物体,设置标签保存,将设置好的图片保存在Annotations文件夹内。
打开voc_classes.txt文件,设置好自己保存的标签名字。如图所示