yolo模型训练工具分享

前言

2026年2月20日，我租下了人生中第一台阿里云服务器。虽然配置仅为2核2G，但已能满足我的需求。起初，我被YOLO模型的训练过程所困扰——不仅需要下载各种依赖、配置复杂环境，还耗费了我整整一上午才勉强搞定。更令人沮丧的是，这套方法换一台电脑就可能失效。此外，训练模块本身也是一大挑战：想要得到一个好模型，必须划分数据集、选择合适的预训练模型，并调整大量参数。整套流程对新手而言，实在不够友好。

于是，我萌生了一个想法：制作一个图形化的YOLO训练界面。最初方案是打包成单个exe文件，用户只需双击即可运行。完成后，我又想：如果能支持远程发布和版本更新，岂不是更方便？然而，当我搭建好服务器并准备上传时，却发现这个exe文件在其他电脑上无法运行。查阅资料后才知道，是因为缺少运行依赖。

随后，我将所有依赖都打包进同一个exe文件。但新问题接踵而至：Windows系统对可执行文件的大小有限制（2G），而我的文件已达1.9G。虽然能够打开，但启动速度极其缓慢。于是，我再次调整思路，转向制作安装包。

在Windows上，我采用Inno Setup制作exe安装包；在Linux上，则采用deb格式制作安装包。最终，我将安装包的大小优化为1.92G（Windows）和2.42G（Linux）。虽然体积较大，但安装后即可直接使用，无需配置环境和依赖，且在不同电脑上都能正常运行。

后来，我开始制作网站来宣传我开发的软件，搭建数据库，并逐步完善网站的各种功能。至今，我的网站已具备下载功能、版本信息查看功能、常见问题解答功能、登录功能、群聊功能、赞助功能以及问题反馈功能。

软件介绍

目前只有模型训练功能和设置是完善的，其他功能还在进一步开发中。
该软件内部会自动识别CPU和显卡，优先使用显卡进行训练，安装包里内置各个版本的预训练模型，可以随时切换选择，内部自动对数据集进行划分（训练集85%，验证集15%）并且支持xml,txt,json这三种格式的数据集，内部会自动识别并进行转换，注意：如果你使用的是yolo官方格式的数据集(.txt格式)则不会进行检查和进行转换，其余格式则会进行检测判断类别名称是否填写完整。

另外，你还可以在设置中切换ui主题选择自己喜欢的样式，你甚至可以上传一张图片作为背景图(图片储存在本地并不会上传到服务器)，你也可以保存当前的训练参数设置以及ui设置，方便你下载训练时可以直接使用。

使用教程

1.在数据集配置栏，选择图片目录和标注目录以及最终模型保存的路径。需要注意的是图片文件夹中只能存放你采集的照片，标注目录文件夹中只能存放你标注好的数据，并且每张图片的名字要和其标注文件的名字相同(除了后缀)。如果你没有填写类别名称直接点击开始训练时，如果你的标注数据格式是xml或者json就会出现如上图那样，软件会提示你输入类别名称，你可以直接复制进去但要保证格式正确（一行一个类别名称），如果你的标注数据格式为txt的话可能就没有这些检查会直接进行训练，最终你得到的模型什么都识别不到。

2.配置完数据集后就可以进行调整模型参数，选择你的预训练模型（也就是想要训练什么版本什么类型的模型）预训练模式是什么这里只说明教程不过多赘述。你可以点击预训练模型那里来进行选择模型，也可以点击自定义按钮选择自己的模型，你也可以将自己的模型复制到软件安装的目录内，这样软件中就会显示出你复制过去的模型，你也可以在安装目录删除不需要的模型。

你可以自己调节训练轮次，轮次越少训练越快，这里默认为2000，你不用担心它训练很慢因为yolo有早停机制，当近100轮训练中模型没有太大的变化时会自己结束训练并将效果最好的模型导出，所以你可以放心的调大训练轮次，这里2000就是一个比较合适的参数。

图片尺寸一般为640或320，320训练的更快640更精确,这里默认为640，如果你是新手可以直接使用这个参数。

批次大小是每轮训练有多少张图片同时进行训练，这个参数直接影响着你的模型训练速度，默认为16，因为这个参数兼容大部分电脑。如果你的显卡很好比如4060你可以直接调整为32，但如果你调得过大可能会崩溃，所以这个参数还是要根据图片数量和电脑配置来做决定

mosaic增强这个参数请自行查找资料，默认为1.0，新手可以直接使用不会有影响。

断点续训，当你训练模型到一半突然电脑关机或者软件崩溃时，你可以将训练了一半的模型继续拿出来训练这极大地节省了时间，但要注意的是你需要在预训练模型那里选择你中断的这个模型，一般是在模型保存路径中train/weights文件夹内名为last.pt，选择这个模型作为预训练模型，然后再勾选断点续训就可以直接点击开始训练啦，模型参数可以不填模型会按照上次的参数配置进行继续训练。注意：数据集配置中的参数还是要重新选择。图片目录和标注目录要和中断时的目录一样。

3.数据集配置和模型参数填写完毕后就可以直接点击开始训练啦，如下图：

模型训练完成后会自动将train文件保存到你填写的模型保存路径，如果训练过程中有报错也会在训练日志中提示以便查明原因。最后你可以在设置中点击保存配置，这样下次打开软件就还是现在的配置。至此，教程结束！

说明

使用此软件需要进行联网（不联网无法使用），用户的图片以及模型等信息不会上传至服务器！服务器只记录用户的累计使用软件时长，只用于统计无其他用途。本软件拥有自更新功能，用户可以在应用内进行更新。本软件支持Windows和Linux(Ubuntu)系统。

如果你需要不联网使用此软件功能或者需要定制的版本可以去官网联系开发者

此软件工作量巨大，本人不仅要维护网站还要维护windows和linux系统软件正常使用，再加上本人正常参与学校竞赛，所以更新的时间很少请见谅。如果你有任何的问题可以前往官方加入群聊或者点击问题反馈功能填写表单进行反馈问题。

官网：yolotool.top 软件下载地址：yolotool.top

以上内容如有说的不对的地方或有侵权行为请指出，谢谢！