1.收集数据集
创建一个dataset文件夹用来存放图片数据集。
我这里使用的图片数据集,是对一段视频进行抽帧得到的200张狗狗图片。

在dataset文件夹下新建images和labels文件夹,并将200张狗狗图片放入images中。

2.标注数据集
2.1安装标注工具labelimg

2.2打开labelimg

2.3点击open dir

2.4选择我们创建好的images

点击选择文件夹后紧接着会弹出新选择,这时候我们选择images同目录下的labels文件夹即可
 2.5标记数据集
 2.5标记数据集
 
选择结束后,右下角会出现数据集的绝对路径,点击第一个开始标记即可

在标记之前需要将保存格式设置为yolo,在view中选择auto save mode可开启自动保存。
 
 

按住w开始画框即可

画完后,对应的labels会生成相对应的标注结果

3.划分数据集
分别在images和lables这两个文件夹中新建文件夹train和val。


同时对images中的图片进行划分,这里我选择了20张图片放到了val文件夹中用于验证。其余放到train文件夹中即可。

同时labels也对应的将标注结果划分到train和val中,数据集构建完成