Mask RCNN使用及实现详解（3）

数据准备

1、 环境配置
标注软件：labelme
labelme有python的版本，可以使用pip 安装，安装命令如下：

pip install labelme

2、 图像标注

安装完成后可以直接在命令行中输入labelme启动，启动后界面如下：

labelme初始界面

点击右上角opentdir图标，选择图片所在的文件夹，打开图片：

opendir按钮

打开图片

然后点击create ploygons图标开始标注：

create polygons图标

标注页面

标注完成后会弹出标签页，选择标注目标对应的标签。

标签选择

标注完成后，会生成一个与图片同名的json文件。

3、 文件处理

json格式的数据并不是mask rcnn需要的数据格式，还需对数据进行转换，转换用的python代码如下：

import os

#
path = r'C:\Users\hrl\Desktop\lightdata\origin'  # path为json文件存放的路径
json_file = os.listdir(path)
os.system("activate labelme")
for file in json_file:
    if ".json" in file:
        os.system("python C:/File/workspace/python/Mask_RCNN/venv/Scripts/labelme_json_to_dataset.exe %s"%(path + '/' + file))

转换后的文件目录如下：

转换后的文件目录

高版本的labelme可能没有yaml文件，解决方法如下：
方案一：
https://blog.csdn.net/winter616/article/details/104426111/
方案二：
yaml文件主要记录的是标签的信息，在label_names.txt中同样记录了标签信息。可以使用label_names.txt代替yaml文件，只是这需要修改Mask rcnn加载数据的代码（该代码在后续文档中介绍）。