AI打游戏-叁(标注图片)

发布时间 2023-09-15 00:49:22作者: 言午日尧耳总

目标

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

步骤

安装标注工具

方式1:Docker

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

方式2:pip安装

# 安装
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"保存



标注

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

导出图片及标注档

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

本节结果

  • 解压后如下目录结构
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