当我们已经准备了大量用于图片标注所需的图片后,便需要使用labelIMG工具进行打标签的操作了。
1、标签规划
在我们开始打标签之前,首先需要注意的是,打标签是我们开展训练、识别工作的基础,我们需要准确最好标签的规划,明确都有哪些标签,并在venv/Lib/site-packages/labelImg/data,找到predefined_classes.txt文件,输入所规划的标签类型。如下图所示。
在这里,笔者要对路面的骑摩托、电动自行车是否佩戴头盔进行识别。因此我们就设置了helmet、nohelmet者两个参数。同时笔者还需要实现和是否加装雨棚的识别,因此便设置了rainshield这个标签。为了避免将穿雨衣识别成加装雨棚,笔者还设置了raincoat这个标签,用来区分。此外,笔者还设置了bike标签,用来避免出现将骑自行车的人员识别成没有戴头盔的骑电动自行车或者摩托车的人员,造成识别的不准确性。
2、下载和安装labelIMG
labelIMG是一个用于给图片添加标注的工具,它可以生成YOLO格式的标签文件,用于训练自定义的目标检测模型。以下是使用LabelImg的教程,希望对你有帮助。
可以使用pip命令来安装LabelImg,也可以从GitHub上下载源代码并运行。假设你已经安装了Python3,你可以在命令行中输入以下命令来安装LabelImg:
pip install labelIm