yolov5训练自己的数据集

下载yolov5源码（网上很多，也很方便下载）需要将train.py文件中的这三个改成自己的。

在这里插入图片描述
首先在yolov5s.yaml文件中
在这里插入图片描述
红框内修改为自己数据集中的类别个数。
在数据集myVOC.yaml文件中将红框内的数据换成自己的数据集路径
在这里插入图片描述
训练之后的模型存储在runs/train/exp中。

版权声明：本文为xuhong_001原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。