前言
本人构建图像分割数据集时,需要对图像进行标注,且得到相应的gt_image,以下是实践过程中的操作,在此记录,供需要者参考。
一、操作步骤
1.下载Labelme(Ubuntu下)
可在终端输入如下代码:(建议在虚拟环境下进行)
# 创建labelme的环境
conda create -n labelme python=3.7
# 激活labelme环境
conda activate labelme
# 安装
conda install pyqt
pip install labelme
#打开labeme工具
labelme
这里可能会出现以下的报错,诸如:
(1) pyqt版本过高:qt.qpa.plugin: Could not load the Qt platform plugin “xcb“ in … 解决方法如下:
pip uninstall pyqt5
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pyqt5==5.14.0
(2) 终端输入"labelme"后闪退:python版本过高(在虚拟环境下设置低版本的应该不会有此报错)
2.使用labelme进行标注
在终端输入labelme后,会弹窗出现标注的界面,如下图所示。
(1)通过左侧菜单栏的Open或者Open Dir,打开你要标注的图片或图片所在目录。一般情况下都需要标注很多图片,因此此处以Open dir为例打开需要标注图片所在的文件夹,在左上角File中选择save automatically。
(2)在上方工具栏“Edit”中选择你要标注的形状(多边形、圆形、矩形等,根据你想标注的区域的形状来选择),此处选择了Create Polygons,在标注过程中,若想撤回已标注的上一个位置,可通过快捷键ctrl+z实现,下图展示了一个标注示例,当你完成一个闭合曲线的标注后,会弹窗显示,让你为当前标注设置一个标签(此处设置成了1),同一张图里可以标注很多的实例。完成该张图片的标注可以进行下一张。
(3)当完成所有图片的标注后,可以看到在其文件夹下生成了对应名称的josn文件,每个json文件记录了图像标注的详细信息。
3.json文件可视化
得到标注的json文件后,可新建一个python文件,通过如下代码将json文件可视化:
#批量处理json文件为二值图
#-*