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标记数据集
选择结束后,右下角会出现数据集的绝对路径,点击第一个开始标记即可
在标记之前需要将保存格式设置为yolo,在view中选择auto save mode可开启自动保存。
按住w开始画框即可
画完后,对应的labels会生成相对应的标注结果
3.划分数据集
分别在images和lables这两个文件夹中新建文件夹train和val。
同时对images中的图片进行划分,这里我选择了20张图片放到了val文件夹中用于验证。其余放到train文件夹中即可。
同时labels也对应的将标注结果划分到train和val中,数据集构建完成