首先我们来看一下这款软件的界面,整体设计简洁直观,大部分参数都已经默认设置好,用户通常只需要调整“批大小”这一项即可,甚至很多情况下连这个都不用改,直接使用默认值也能顺利运行。
接下来进入正题,教你如何在Windows上快速完成自定义数据集的训练,全程无需编写代码、无需手动修改配置文件、也不需要深入了解YOLO背后的复杂原理,小白也能轻松上手!
第一步:[未雨绸缪] 标注你的数据集
我们使用软件内置的 labelImg 工具进行图像标注。该工具支持两种主流格式:Pascal VOC(XML) 和 YOLO(TXT)。
-
Pascal VOC 格式说明:
每张图片对应一个文件,存储标注信息。结构如下:
- :图片文件名
- :图片完整路径
- 中包含 和 ,表示图像分辨率
- 每个目标由一个 描述,包含:
- :类别名称
- :是否为难以识别的目标(可忽略)
- :边界框坐标,包括 , , , ,单位为像素,坐标系原点在左上角
-
YOLO TXT 格式说明:
每张图片对应一个文件,每行代表一个目标,格式为:
所有数值均为归一化后的浮点数(相对于图像宽高),其中 是目标中心点。
建议标注数量: 初步测试:100张以上 效果良好:1000张起 高精度需求:5000张以上 极致效果:10000+ 张
第二步:[霸气登场] 开始训练自己的模型
准备好标注数据后,按照以下步骤操作:
- 图片必须放在名为 JPEGImages 的文件夹中。
- 若使用VOC格式,标注文件(XML)建议放在 Annotations 文件夹中。
- 若使用YOLO格式,标注文件(TXT)必须放在名为 labels 的文件夹中,并与 JPEGImages 同级。
- 在软件中分别导入 JPEGImages 和标注文件夹(Annotations 或 labels)。
- 根据你的显卡显存设置 batch size(批大小),以下是常见配置参考:
- 点击“配置数据集”,软件将自动处理数据路径、生成训练所需文件。
- 数据量小(如2000张):几秒完成
- 数据量大(如1万+):可能需要几分钟
- 配置完成后,点击“获取命令”可查看底层执行指令,或直接点击“开始训练”一键启动!
第三步:[谁与争锋] 模型测试功能全开
这款软件的强大之处不仅在于训练简单,还内置了完整的测试模块,支持全图形化操作:
- 支持上传 图片、视频 文件进行检测
- 支持调用 摄像头 实时检测
- 无需敲命令,鼠标一点即可运行
如果你想了解背后的命令行逻辑,可以选择“外部终端测试”模式,软件会自动生成完整的执行命令供你学习和复用。
️ 注意事项:
在测试时,请确保
文件中的 路径使用的是
绝对路径,相对路径可能导致加载失败。
更多实用功能加持:
- 标注检查功能:自动扫描异常标注,防止错误数据影响训练效果
- 类别查找功能:快速定位某类别的所有标注,避免类别错标、漏标
- 多模型支持:涵盖 yolov3、yolov3-spp、yolov3-tiny、yolov4、yolov4-tiny、yolov7、yolov7-tiny、yolov7x 等主流版本
学习资源推荐:
- 1.2.2版本使用教程:yolov3快速训练助手1.2.2使用教程_哔哩哔哩_bilibili
- 2.0新版UI教程:视频去哪了呢?_哔哩哔哩_bilibili
- 2.1支持YOLOv7:yolov3 yolov4 yolov7训练助手2.1支持yolov7一键训练全程不用写代码傻瓜式操作_哔哩哔哩_bilibili
本次分享就到这里,下期将继续为大家带来更多高效、易用的AI训练工具!