• AI打游戏-叁(bilibili)

目标

• 安装标注工具
• 标注游戏图片
• 导出到本地目录当做训练集

步骤

安装标注工具

• 标注工具LabelStudio

方式1:Docker

• 推荐用这种方式
• 安装后访问地址：http://localhost:8080/

docker run --name label-studio -itd -p 8080:8080 heartexlabs/label-studio:latest

方式2:pip安装

• 启动python环境
• 使用pip安装
• 访问依然是：http://localhost:8080/

# 安装
pip install -i https://mirrors.aliyun.com/pypi/simple/ label-studio

# 启动
label-studio start

创建项目

• 打开LabelStudio，直接注册一个账号
  • 注册使用邮箱，但是不校验，随便写个邮箱格式就行
• 登录后创建项目，创建项目有三个选项
  • Project Name: 项目名称
  • Data Import: 上传图片数据
    • 上传图片有100张的限制，分几次上传几次即可
  • Labeling Setup: 标注信息设置
    • 选择"Object Detection with Bounding Boxes"
    • 设置标签，左侧写入"玩家/武士/地狱犬/萤火虫",然后添加到右侧
    • 右侧的"Labels"，删除默认的标签，点击标签可以设置颜色（将四个标签设置成差别较大的颜色）
• 最后点击"Save"保存

2_0_labelstudio-import.png
2_1_labelstudio-boxes.png
2_2_labelstudio-label.png

标注

• 点击刚才创建的项目，进入标注
• 图片下方是设置的标签，点击后在画面上拖动，将元素框选起来
  • 标签边上的数字是快捷键
• 完成一张后点击右下方的"Submit"保存标注信息
• 点击下一张继续标注

2_3_labelstudio-labeling.png

导出图片及标注档

• 在项目的图片列表中，点击右上角"Export"导出数据
• 格式选择"YOLO"，然后"Export"下载
• 之后解压即可

2_4_labelstudio-export.png

本节结果

• 解压后如下目录结构

project-xxx
│
│  classes.txt  # 每行一个label，从0开始（这里排序可能和设置不同）
│  notes.json   # label的json格式
│
├─images
│      58ffa2e4-357.jpg     # 所有图片
│      aa23ddh4-245.jpg
│
└─labels
        58ffa2e4-357.txt    # 图片对应的YOLO格式标注档
        aa23ddh4-245.txt