制作自己的训练数据集之图像标注工具labelimg和labelme

时间:2024-03-31 14:08:15

第在深度学习中,学会制作自己的数据集然后进行模型训练是一个很重要的步骤,这里教大家下载安装使用两种制作自己的数据集的标注工具labelimg和labelme。

前言

首先简单介绍一下这两种工具,labelimg工具主要用于在目标检测,而 labelme工具图像分割领域制作自己的数据集,我们可以用labelimg用于制作自己的数据集训练像YOLOv3和YOLOv4目标检测模型,而用labelme可以自己制作像MaskRCNN模型做图像分割训练需要的数据集。

目标检测如下图:

制作自己的训练数据集之图像标注工具labelimg和labelme

目标检测和图像分割如下图:

制作自己的训练数据集之图像标注工具labelimg和labelme

话不多说正式开始下载安装使用。

labelimg

labelimg是采用Python语言编写的,界面用Qt设计的,所以首先就是安装pyqt5。

第一步:打开Anaconda Prompt 首先创建一个用于标注数据的py35单独环境名为label的环境,当然也可以用 Anaconda Navigator 界面程序来创建。

制作自己的训练数据集之图像标注工具labelimg和labelme

第二步:查看创建环境是否成功并**环境

制作自己的训练数据集之图像标注工具labelimg和labelme

第三步:添加镜像源下载并安装Pyqt5和labelimg,这里使用豆瓣源

pip install pyqt5 -i https://pypi.douban.com/simple/

制作自己的训练数据集之图像标注工具labelimg和labelme

pip install labelimg -i https://pypi.douban.com/simple/

制作自己的训练数据集之图像标注工具labelimg和labelme

第四步:安装完后,输入命令labelimg,即可打开标注工具界面进行数据标注

制作自己的训练数据集之图像标注工具labelimg和labelme

制作自己的训练数据集之图像标注工具labelimg和labelme

第五步:打开你想要标注的数据集图片进行标注即可,记住每标记完一张记得保存,具体操作如下:

制作自己的训练数据集之图像标注工具labelimg和labelme

在我们进行标注时,我们可以在Anaconda Prompt中实时查看到标注情况。

制作自己的训练数据集之图像标注工具labelimg和labelme

第六步:标注完所有图像之后就可以查看标记结果了,注意这里我们进行标注时有两种标记数据的格式一种是PascalVoc,另外一种是YOLO,这里以常见的VOC数据集格式进行标注,最终得到以xml格式的标注数据。

制作自己的训练数据集之图像标注工具labelimg和labelme

Voc数据集的数据格式,包含对象名,标记框的基本信息。

制作自己的训练数据集之图像标注工具labelimg和labelme

至此,用labelimg制作数据集的过程到此结束,就可以去训练YOLO的模型了。

labelme

labelme安装过程和labelimg大致都是一样,只需把上述第三步不换成pip install labelme -i https://pypi.douban.com/simple/。这两个工具可以同时存在一个环境之中,也就是说如果已经成功安装上其中一个,只需执行第三步即可成功安装另外一个,没有安装的话,简要步骤如下:

1.Anaconda Prompt中创建一个环境

conda create --name=labelImg python=3.6

2.**进入刚建立的新环境,

conda activate labelImg

3.安装界面支持pyqt5包

pip install pyqt5 -i https://pypi.douban.com/simple/

4.下载安装labelme

pip install labelme -i https://pypi.douban.com/simple/

制作自己的训练数据集之图像标注工具labelimg和labelme

5.输入命令labelme,就可以启动程序进行数据标注

制作自己的训练数据集之图像标注工具labelimg和labelme

如图所示,我们可以进行矩形标注、点标注、以及圆形标注等多种格式进行标注,界面的上的按钮的作用和labelimg类似,这里不再赘述。

制作自己的训练数据集之图像标注工具labelimg和labelme

相对于labelimg,我们可以以多种形式进行数据标注,如下图

制作自己的训练数据集之图像标注工具labelimg和labelme

标注完后进行保存save,图片上标注的信息将以json文格式将进行保存,如下图:

制作自己的训练数据集之图像标注工具labelimg和labelme


至此用至此,用labelimg制作数据集的过程到此结束 !

总结:

labelimg标注完数据之后以.xml数据的格式进行保存,只能创建矩形框更加适用于目标检测制作数据集,如YOLO,而labelme以.json文件的数据格式进行保存,可以创建多种类型的边框进行标注,适用于图像分割,如MaskRcnn。

 ·END·

微信号:码农的后花园

制作自己的训练数据集之图像标注工具labelimg和labelme

长按识别关注哦

整理不易,点个好看再走呗

制作自己的训练数据集之图像标注工具labelimg和labelme