一、制作数据集

使用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

二、训练

最后修改日期: 2019年2月24日

作者

留言

撰写回覆或留言

发布留言必须填写的电子邮件地址不会公开。