使用教程

从装好软件到训出能用的模型,一步步走完。整条流程完全在本机离线完成,图片和数据不会离开你的电脑。

01 下载与安装

到下载页选国内(Gitee)或 GitHub 的最新版本,下载 Windows 安装包(.msi / .exe)。双击安装,或解压后直接运行。

02 装训练环境(只需一次)

第一次要训练前,进入设置页点一键安装训练环境。软件会自动检测显卡、选择匹配的 PyTorch 版本,并用国内镜像加速下载。

提示:训练环境是独立的,不会污染系统里其它 Python。换电脑重装软件后再装一次即可。

03 新建项目

打开软件点新建项目,选一个空文件夹并起名——一个项目就是一个文件夹,图片、标注、模型、训练记录都存在里面,整个拷走即可备份或换机。

之后从最近项目可以快速打开。

04 导入数据

进入数据页,把图片、整个文件夹直接拖进来,或点导入选择。

巡检场景示例图片
示例:导入的巡检场景图片——消防通道违停、电动车进电梯、未戴安全帽、宠物入内、吸烟、烟雾火情等。

05 标注

进入标注页,在图上拖拽画框,给框选类别。常用操作:

标注示意:画框并给框选类别
标注示意:在图上画框、给每个框选类别。图中一张工地照同时标了「未戴安全帽」(红)和「安全帽」(绿)两个类别。

省力思路:先手标一小批(几十到一两百张),就足够训练出一个初版模型,之后用它来预标注剩下的图,你只要改错,越往后越快。

06 数据体检

进入体检页,一键扫描数据里的常见问题:

07 划分数据集

在数据页把图片分成训练 / 验证 / 测试。验证集用来在训练中看指标、挑最好的模型,测试集用来最后客观评估。常用 8:1:1 或 7:2:1。

08 训练

进入训练页,选模型骨架开练:

09 评估

进入评估页看模型到底行不行:

10 模型预标注

回到标注页,用训好的模型自动标图:

模型预标注示意
预标注示意:模型自动把目标框出来,你只需审核、改掉少数错的。图为「违停车辆」检测结果。

11 导出与部署

进入部署页,把成果用起来:

12 反馈问题

用着有问题、有建议,点软件右上角的反馈,写下描述、需要时附上截图提交,会直接送到作者邮箱。留个联系方式方便回复。

也可以到开源仓库提 Issue。感谢每一条反馈 —— 小物学长会看。


准备好了?下载 Wulabel,开始标你的第一张图。