这篇博客记录的是我制作数据集的过程和一些心得,希望可以帮助到一些小伙伴。
1.首先,我们要知道yolo数据集的格式如下:
其中我们做数据集的时候一定要注意!label里面的txt文档!一定要是txt的格式才是yolo可以跑!如果不是txt格式的,要去改下格式。可以去打标签的软件里面再重新打标,有点麻烦,不过可行。
把自己要训练的照片放在train里面,然后放几张测试集的照片放在val里面。
打好标签之后就可以把标签按照train和val分类,把对应的标签放在对应的标签文件夹里面。
在labels文件夹里面还需要新建一个class.txt文档用来写训练的种类。
比如:patato,tomato,apple......
或者可以参考这个博主写的,用代码自动划分数据集和测试集:
在data目录下创建程序 split_train_val.py
并