用yolo做目标检测任务时,网上的一些标注软件如Labelimg,anylabel等的bug太多,而且功能比较少,比较不方便,因此,本人自己开发了一个专门用于YOLO和YOLO-OBB的标注软件,集成了识别和训练功能。
(下载链接在文章末尾)
本文介绍的是一种用于YOLO目标识别的标注软件,该软件可以用于水平矩形框、任意角度的矩形框和任意多边形框的标注,而且带有识别和训练功能,再也不需要自己去跑代码训练了,本软件同时还有自动分配功能,可以自动分配好训练集和验证集,同时还可以往训练集里添加背景图片。
下面我介绍一下软件的主要功能:
图1 JieLabel软件主界面
图1为软件的主界面,主要分为四个区域,最左边是功能区域,有选择图片、画框、自动分配、训练和识别等功能;中间是显示图像区域,可以对图像进行放大缩小和旋转操作。右侧上方为标签显示区域,左边的是当前图像标签显示区域,右边的为已创建的类别显示区域,会按顺序显示出已经创建的类别,右侧下方为图片地址显示区域,用于显示所有图像的地址。
左侧按键区介绍:
1)打开图片(文件夹):此按键可以打开单张图片或选择文件夹内的所有图片,单击此按钮会弹出选择地址的窗口,选择即可。
2)画框:此按键为画框按键,单击鼠标左键即可实现画矩形框(水平的或者斜的都可以画),画框完毕会弹出命名标签对话框,此按键快捷方式为w键。如下图
3)放大、缩小、+-90°:这四个按键会对当前图像进行放大缩小以及旋转操作,放大缩小只是改变显示方式,不改变图像真实大小,而旋转会改变图像的真实方向,放大缩小快捷方式为ctrl+鼠标滚轮,ctrl+滚轮上滑为放大、下滑为缩小。
4)翻页(上下张):此按键在打开文件夹方式下可以对更换当前图像,快捷键为Q、E。当然,单击右侧图片地址也可以进行图像的更换。
5)标签数目:此按键会显示当前标签类别的个数,可以让大家更直观的看到以及标的类别有多少个。
6)进出背景:单击此按键,会把当前图像放入或拿出背景,被选入背景的图像会被放入训练集,且不会有标签。放入背景的图像在右侧图像地址区域以蓝底色显示
7)自动保存:勾选此框,在进行更换图像等操作时会自动保存标注文件。
8)标注模式:选择会更改标注方式,包括水平矩形、斜矩形和任意多边形。
9)随机分配:此按键可以把已经标注的图像(包含背景)进行按比例自行分配,按YOLO的标准生成训练需要的数据集(包含v5版本和v8版本),包含训练集train和验证机val,以及配置文件xxx.yaml。
训练时可以直接使用的,如果使用yolo官方的代码进行训练,只需data参数选择本文数据集配置文件地址即可。
10)预识别:此功能用于半自动标注,例如你需要标注500张图片,你可以先标注100张进行训练,得到一个权重文件,然后使用其功能进行识别,软件会按识别结果生成标注文件,你只需要微调图像即可,不需要重新标注,可以大大降低标注工作量。
11)训练:此功能用于训练模型,单击此按钮弹出训练窗口,如下图。
该图像左侧会显示Map随epoch的变化而变化曲线,右侧则是训练的参数设置,包含选择预训练权重文件、cfg、data、等等,以及超参数设置如下图。
当设置完成后单击开始按钮就可以开始训练了,同时终端窗口也会打印训练进度。训练过程是独立进行的,因此在训练的同时,你也可以进行图像标注等工作。
有问题进交流群:
http://qm.qq.com/q/nVIq6FVyOkhttp://qm.qq.com/q/nVIq6FVyOk
具体快捷键和其他功能:
4.5.0版本更新日志和下载链接(09-26)