AI打游戏-叁(标注图片)
目标
- 安装标注工具
- 标注游戏图片
- 导出到本地目录当做训练集
步骤
安装标注工具
- 标注工具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"保存
标注
- 点击刚才创建的项目,进入标注
- 图片下方是设置的标签,点击后在画面上拖动,将元素框选起来
- 标签边上的数字是快捷键
- 完成一张后点击右下方的"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