结合 mmdetection 、 label studio 实现数据集自动标注、模型自动迭代的 AI 闭环。
软件 | 版本 |
---|---|
python | 3.9.12 |
label-studio | 1.6.0 |
label-studio-ml | 1.0.8 |
label-studio-tools | 0.0.1 |
torch | 1.11.0+cu113 |
torchaudio | 0.11.0+cu113 |
torchvision | 0.12.0+cu113 |
mmcv-full | 1.5.1 |
mmdet | 2.24.1 |
LabelStudio
├── backend // 后端功能
│ ├── examples // label studio ml 官方示例(非必须)
│ ├── mmdetection // mmdetection 人脸检测模型
│ ├── model // label studio ml 生成的后端服务 (自动生成)
│ ├── workdir // 模型训练时工作目录
│ | ├── fcos_common_base.pth // 后端模型基础权重文件
│ | └── latest.pth // 后端模型最新权重文件
│ └── runbackend.bat // 生成并启动后端服务的脚本文件
├── dataset // 实验所用数据集(非必须)
├── label_studio.sqlite3 // label studio 数据库文件
├── media
│ ├── export
│ └── upload // 上传的待标记数据集
└── run.bat // 启动 label studio 的脚本文件(非必须)
在根目录下执行:
label-studio --data-dir . -p 80
- 修改配置文件
修改backend/mmdetection/conf.yaml
和backend/model/conf.yaml
文件中路径为你自己的路径:
config_file: D:\CommonProject\LabelStudio\backend\mmdetection\model\fcos_common.py
checkpoint_file: D:\CommonProject\LabelStudio\backend\workdir\latest.pth
device: cuda:0
workdir: D:\CommonProject\LabelStudio\backend\workdir
- 添加预标记模型权重
把与label_studio_backend.py
中模型匹配的权重文件复制到 backend/workdir/latest.pth 中。 - 启动后端服务
在 backend 目录下执行:
label-studio-ml start model --host 0.0.0.0 -p 8888