用户在采集完用于训练、评估和预测的图片之后,可使用精灵数据标注工具完成数据标注。精灵标注支持在Windows/macOS/Linux三个系统上使用,Mac的话可以到MacStore中搜索colabeler下载即可。
开始标注自定义数据集之前,可以先预览精灵标注官网的文字教程和视频教程。
- 开始标注
打开精灵标注工具。
请按照下述步骤标注数据集:
(1) 点击新建
,然后选择位置标注
,选择图片文件夹
,修改填写所需的分类值
(注意:以英文逗号隔开),点击创建
按钮,软件会自动加载文件夹下的图片(png,jpg,gif)并创建一个项目。
位置标注支持三种类型:矩形,多边形和曲线。选择简单好用的多边形框
,沿着目标的边缘画多边形,完成后在右侧输入目标的类别。
注意:切记单张图片标注完成后进行保存,点击下方中央的勾按钮或者使用快捷键ctrl+s。
然后可以点击左边的前一个后一个或者直接使用键盘的向左按钮和向右按钮来切换图片。
(2) 单击目标框,鼠标拖动可以整体移动多边形的位置;点击左侧的删除选框
可以删除画错的目标框;点击右侧的标注信息
可修改目标类别。请根据自己的需要执行这一步骤,若不需要修改,可跳过。
(3) 当所有图片的标注都完成后,点击左侧的导出
,输出方式选择JSON
,指定保存位置
,点击确定导出
保存所有图片的标注文件。
注意:导出的标注文件位于保存位置
下的outputs
目录。
精灵标注产出的真值文件可参考我们给出的文件夹docs/annotation/jingling_demo
注意: 对于中间有空洞的目标(例如游泳圈),暂不支持对空洞部分的标注。如有需要,可借助labelme。
最后用我们提供的数据转换脚本将上述标注工具产出的数据格式转换为模型训练时所需的数据格式。
- 经过数据格式转换后的数据集目录结构如下:
my_dataset # 根目录
|-- outputs # 标注工具导出目录
| |-- annotations # 数据集真值
| |-- xxx.png # 像素级别的真值信息
| |...
| |-- class_names.txt # 数据集的类别名称
| |-- xxx.json # 标注json文件
|-- xxx.jpg(png or other) # 数据集原图
|-- ...
- 运行以下代码,将标注后的数据转换成满足以上格式的数据集:
python pdseg/tools/jingling2seg.py <PATH/TO/LABEL_JSON_FILE>
其中,<PATH/TO/LABEL_JSON_FILE>
为精灵标注产出的json文件所在文件夹的目录,一般为精灵工具使用(3)中保存位置
下的outputs
目录。
我们已内置了一个标注的示例,可运行以下代码进行体验:
python pdseg/tools/jingling2seg.py docs/annotation/jingling_demo/outputs/
转换得到的数据集可参考我们给出的文件夹docs/annotation/jingling_demo。其中,文件class_names.txt
是数据集中所有标注类别的名称,包含背景类;文件夹annotations
保存的是各图片的像素级别的真值信息,背景类_background_
对应为0,其它目标类别从1开始递增,至多为255。