制作yolov5数据集

53 阅读1分钟

准备自己的数据集

Fruit26数据集

创建Fruit25文件夹,用于一会儿保存划分好的训练集、测试集和验证集

split.py

通过split.py来将标注好的图片划分为训练集、测试集和验证集

编写yaml文件

image.png 设置好路径和类别

image.png 运行过程中,会自行去yaml文件中设置的目录下寻找图片和标签