本栏目分为6个部分:
- 第1部分:Detectron框架简介与安装
- 第2部分(本文):使用LabelImgPlus对图片进行标注
- 第3部分:数据集处理
- 第4部分:在detectron上部署数据集
- 第5部分:训练
- 第6部分:测试
GitHub: https://github.com/zhiweichen95/detectron_my
对于自行采集的图片做目标检测任务,都需要对每张图片进行人工标注,使用软件工具能够大大节约时间并且容易的把图片中需要检测的目标整理下来。本章节对LabelImgPlus标注工具从安装到使用进行详细介绍。
用手机拍摄150张的“人骑自行车”和150张的“人骑电动自行车照片”,利用目标检测技术,检测出“人”,“自行车”, “电动自行车”.
LabelImg is a graphical image annotation tool. It is written in Python and uses Qt for its graphical interface. The annotation file will be saved as an XML file. The annotation format is PASCAL VOC format, and the format is the same as ImageNet.
本人使用的是mac,LabelImgPlus 也有mac的安装版本,但是用起来鼠标会存在延迟,所以安装了Ubuntu系统(虚拟机)进行标注,使用win系统的读者也是可以安装一个Ubuntu系统来进行标注。这个软件在win系统下安装也是比较麻烦的,本人建议安装Ubuntu虚拟机安装LabelImgPlus 。
本小节也是详述LabelImgPlus 在Ubuntu系统的安装步骤。如各位读者使用mac或者win系统安装遇到问题也欢迎留言交流。
1 Download project source
$ git clone https://github.com/lzx1413/labelImgPlus
2 Check support and install dependent libraries
Ubuntu (using python2.7)
$ sudo apt-get install pyqt4-dev-tools
$ pip install lxml
$ pip install qdarkstyle
$ pip install numpy
$ pip install Pillow
$ pip install opencv-python
注:pip 安装时提示没有安装权限的时候,执行sudo pip install ×××× 指令
到此,LabelImgPlus 安装完成。
$ cd labelImgPlus
$ ./labelImg.py
PATH/labelImgPlus/data/
文件夹中,找到predefined_classes.txt
,本项目需要检测people, electric bicycle, bicycle
三个类别,所以需改成如下格式:Open Dir
,打开存放图片的文件夹。即可把图片载入。Create RectBox
开始标注,框好物体后显示物体列表,直接选择列表中对应的类别即可。 本人标注完成后,对数据集额外处理了超久,如:文件名修改、文件名修改了自然xml文件也要对应修改等等。加上使用detectron框架数据要放在一个文件夹内,不同种类分别命名会比较好。看到这里,也许各位读者还是不明白为啥要这么做,那您先跟着做吧~看到后面的训练部分,自然会明白了!现就一些文件命名坑阐述如下:
本项目需要检测people, electric bicycle, bicycle
三个类别,人骑自行车150张图片,人骑电动自行车150张图片。
bicycle001.jpg
bicycle002.jpg
bicycle003.jpg
......
bicycle150.jpg
electric_bicycle001.jpg
electric_bicycle002.jpg
electric_bicycle003.jpg
......
electric_bicycle150.jpg
全部标注完成后,总文件数为150*2+150*2=600
个。分别为:
bicycle001.jpg
bicycle002.jpg
bicycle003.jpg
......
bicycle150.jpg
bicycle001.xml
bicycle002.xml
......
bicycle150.xml
electric_bicycle001.jpg
electric_bicycle002.jpg
electric_bicycle003.jpg
......
electric_bicycle150.jpg
electric_bicycle001.xml
electric_bicycle001.xml
......
electric_bicycle150.xml