【yolov5目标检测】使用yolov5训练自己的训练集

2023-10-23 19:29:09 浏览数 (1)

数据集准备

首先得准备好数据集,你的数据集至少包含images和labels,严格来说你的images应该包含训练集train、验证集val和测试集test,不过为了简单说明使用步骤,其中test可以不要,val和train可以用同一个,因此我这里只用了一个images

其中images装的是图片数据,labels装的是与图片一一对应同名的yolo格式txt,即类别号,经过归一化的中心x和y坐标以及宽和高

下载yolov5

到GitHub上下载整个项目的压缩包zip下来

 下来解压zip,把我们刚刚的数据集也放进去

再下载一个yolov5的预训练模型,我这里选择yolov5n.pt,下下来也放到解压文件夹中

然后用pycharm打开这个文件夹,看看哪里标红还差什么软件包没安装给安装上

配置yaml

先配置一下yolov5预训练模型的yaml,我下载的是yolov5n.pt模型,因此需要配置一下yolov5n.yaml,修改nc的数值为类别数目,我这里的数据集只有乌骨鸡和狮头鹅,因此改成2

代码语言:javascript复制
# YOLOv5 


	

0 人点赞