使用LabelImg进行图像标注

计算机视觉领域,图像标注是一项基础且重要的工作。LabelImg是一个简单易用、轻量级、与YOLOv5兼容且免费的图像标注工具。本文将指导如何安装并使用LabelImg来为图像数据集创建标签。这个过程可能会花费数小时的时间,因此请确保有足够的耐心和专注力。

安装LabelImg

可以通过多种方式安装LabelImg。首先,需要克隆它的仓库:

git clone [repository-url]

然后在终端窗口中运行以下命令进行安装:

Ubuntu:

cd path/to/dir/LabelImg sudo apt-get install pyqt5-dev-tools sudo pip3 install -r requirements/requirements-linux-python3.txt make qt5py3

MacOS:(首先导航到LabelImg目录)

brew install qt pip3 install pyqt5 lxml make qt5py3

Windows + Anaconda:(首先导航到LabelImg目录)

conda install pyqt=5 conda install -c anaconda lxml pyrcc5 -o libs/resources.py resources.qrc

安装完成后,需要编辑预定义的自定义类别。导航到labelImg/data目录并寻找predefined_classes.txt文件。打开它并输入希望在应用中使用的标签。对于口罩检测项目,标签应该如下所示:

开始标注图像

在开始标注过程之前,打开文件浏览器并创建一个文件夹来保存已经增强的图像。在该目录中解压缩在数据增强步骤中获得的文件,并在其中创建一个名为Labels的文件夹,将在这里保存LabelImg生成的所有文本文件。

现在,通过运行以下命令启动应用程序:

python3 labelImg.py

应用程序将启动。现在让开始图像标注。这需要大量的耐心,但可以完成它!

从图像生成训练数据集

在开始标注过程之前,打开文件资源管理器并创建一个文件夹来保存将要保存的图像。解压缩在数据增强步骤中获得的文件,并在其中创建一个名为Labels的文件夹,将在这里保存LabelImg生成的所有文本文件。

是时候标注图像了。打开LabelImg,点击左侧面板上的Open Dir并选择包含图像数据集的文件夹。要选择标签存储的位置,点击Change Save Dir并选择上面创建的新的Labels文件夹。完成后,将看到文件夹中的第一张图像:

在开始标注过程之前,请确保在左侧面板上选择了"YOLO"模式:

点击Create RectBox并围绕图像中的每个脸部绘制一个正方形。尽量覆盖每个脸部的尽可能多的部分,同时不要在正方形内留下大量空白。每次绘制一个正方形,都会弹出一个提示框,要求选择与正方形对应的标签:

在过程结束时,应该得到一张完全标注的图像:

注意右侧的标注摘要。使用它来跟踪在每张图像上所做的工作。一旦准备好了,点击左侧面板上的Save,然后点击Next Image。

经过几个小时的乐趣后,超过4,000张图像的数据集被标注完毕。生成的g.txt文件包含什么内容?让检查其中的一个:

沪ICP备2024098111号-1
上海秋旦网络科技中心:上海市奉贤区金大公路8218号1幢 联系电话:17898875485