想要进行模型训练,就需要现有标注好的数据。
有时候要训练的模型在网上找不到合适的免费或付费样本,这时候就需要我们自己动手来完成数据标注。
最常用的工具就是Labelme。labelme是一款开源的图像标注工具,支持多边形、矩形、圆形等形状的标注,适用于目标检测、语义分割等任务。
1.软件安装
基础环境建议使用anaconda创建一个独立的环境,避免依赖冲突。
使用IDE软件终端(我用的是pycharm)输入命令行:pip install labelme,坐等安装完成。
在这个过程中,pyqt5等需要的库也会一并自动装进去,可以说非常方便了。
安装完成后,运行labelme --version查看版本信息,确保安装成功。
2.数据标注
启动labelme的方式也很简单:激活anaconda环境后,运行labelme命令,即可打开主界面。
上方的open按钮是打开一张图,opendir是打开文件夹(如果有一大堆照片用来标注,可以选择这个)。
选择完成后,软件会自动加载图像。
选择create polygon按钮,在图像上多边形绘制出目标区域——绘制完成后软件自动弹窗,让用户选择当前构绘区域的所属类别。
图片完成后,点击保存,选择存储路径,形成标签json文件: