一、制作数据集
使用labelImg
(1)创建文件夹
按照VOC数据集的要求,创建以下文件夹
Annotations:用于存放标注后的xml文件
ImageSets/Main:用于存放训练集、测试集、验收集的文件列表
JPEGImages:用于存放原始图像
(2)标注数据
将图片集放在JPEGImages文件夹里面,注意图片的格式必须是jpg格式的。
打开labelImg标注工具,然后点击左侧的工具栏“Open Dir”按钮,选择刚才的JPEGImages文件夹。这时,主界面将会自动加载第一张照片。
(3)划分训练集、测试集、验证集
完成所有照片的标注后,还要将数据集划分下训练集、测试集和验证集。
在github上下载一个自动划分的脚本(https://github.com/EddyGao/make_VOC2007/blob/master/make_main_txt.py)
然后执行以下代码
python make_main_txt.py
留言