Yolov5 模型训练

174 阅读1分钟

使用 Label Studio 进行图片标记

安装运行 Label Studio

新建一个 Conda profile,然后按照 Label Studio 官网的命令安装:

conda create -n label python=3.10
conda activate label
pip install -U label-studio

运行 Label Studio 就是切换到这个 profile 下,然后执行:

label-studio

Label Studio 是一个 BS 架构程序,上述命令会启动它的 Server 端,通过浏览器访问 http://localhost:8080 即可显示其界面。

首次打开可以在 Sign Up 页面上注册用户,这个注册就是本地注册不联网。注册完毕后以后都要用这个用户进行登录,这样可以看到历史的标注项目。

注意,使用过程中全程不可以关闭命令窗口。Label Studio 的数据默认情况下是存放在用户 Home 目录中的(Windows 为 C:\Users\<USER>\AppData\Local\label-studio\),所以关闭程序再重新打开不会丢失数据。

Yolov5 格式标记

点击右上方 Create 按钮,创建标注项目。

  1. 给项目起个名字,如有必要再增加描述;
  2. Data Import 可以先不做,后面再上传文件;
  3. Labeling Setup 要选择 Computer Vision 类别中的 Object Detection with Bounding Boxes;
  4. 其它设置不用改,把标注要用到的 Label 设置上。已有的 Plane、Car 可以删除掉。点击 Label 名称可以为其设置颜色,建议用不同颜色区分。